fix: use explicit list of platform names for linux environmental vari…

…able assignment logic in _get_mpi_env
brown-ccv · Aug 30, 2024 · 31159a7 · 31159a7
1 parent d27ee78
commit 31159a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: