remove setting ENVs during import

flatironinstitute · May 3, 2024 · d888caa · d888caa
1 parent c9def7f
commit d888caa
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/inferelator/__init__.py b/inferelator/__init__.py
@@ -1,11 +1,3 @@
-# Set threading control variables if they're not already set
-
-import os
-os.environ["MKL_NUM_THREADS"] = os.environ.get("MKL_NUM_THREADS", "1")
-os.environ["NUMEXPR_NUM_THREADS"] = os.environ.get("NUMEXPR_NUM_THREADS", "1")
-os.environ["OMP_NUM_THREADS"] = os.environ.get("OMP_NUM_THREADS", "1")
-os.environ["OPENBLAS_NUM_THREADS"] = os.environ.get("OPENBLAS_NUM_THREADS", "1")
-
 from inferelator.workflow import inferelator_workflow
 from inferelator.crossvalidation_workflow import CrossValidationManager
 from inferelator.utils import inferelator_verbose_level

diff --git a/inferelator/distributed/dask_cluster_controller.py b/inferelator/distributed/dask_cluster_controller.py
@@ -29,7 +29,7 @@
     'export MKL_NUM_THREADS={t}',
     'export OPENBLAS_NUM_THREADS={t}',
     'export NUMEXPR_NUM_THREADS={t}'
-    ]
+]
 
 _DEFAULT_CONTROLLER_EXTRA = [
     '--nodes 1',