From 31159a76d705d818fbd111dadab097bd74175dee Mon Sep 17 00:00:00 2001
From: George Dang <53052793+gtdang@users.noreply.github.com>
Date: Fri, 30 Aug 2024 14:35:28 -0400
Subject: [PATCH] fix: use explicit list of platform names for linux
 environmental variable assignment logic in _get_mpi_env

---
 hnn_core/parallel_backends.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hnn_core/parallel_backends.py b/hnn_core/parallel_backends.py
index da02e89eb..3dda3b7cc 100644
--- a/hnn_core/parallel_backends.py
+++ b/hnn_core/parallel_backends.py
@@ -84,7 +84,8 @@ def _gather_trial_data(sim_data, net, n_trials, postproc):
 def _get_mpi_env():
     """Set some MPI environment variables."""
     my_env = os.environ.copy()
-    if 'win' not in sys.platform:
+    # For Linux systems
+    if sys.platform not in ['win32', 'darwin']:
         my_env["OMPI_MCA_btl_base_warn_component_unused"] = '0'
 
     if 'darwin' in sys.platform: