move it to the models init

twallema · Aug 30, 2024 · c8b052b · c8b052b
1 parent eb70e14
commit c8b052b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 17 deletions.
diff --git a/src/pySODM/__init__.py b/src/pySODM/__init__.py
@@ -1,17 +0,0 @@
-"""
-By setting the following environment variables, pySODM prohibits Numpy, Torch & Tensorflow from automatically multithreading large matrix operations
-This does not play nice with the pySODM feature of running the model N times in parallel using multiprocessing
-"""
-
-import os
-
-# Numpy & Torch
-os.environ['VECLIB_MAXIMUM_THREADS'] = '1'
-os.environ['OPENBLAS_NUM_THREADS'] = '1'
-os.environ['MKL_NUM_THREADS'] = '1'
-os.environ['BLIS_NUM_THREADS'] = '1'
-os.environ['NUMEXPR_NUM_THREADS'] = '1'
-
-# Tensorflow
-os.environ['TF_NUM_INTEROP_THREADS'] = '1'
-os.environ['TF_NUM_INTRAOP_THREADS'] = '1'

diff --git a/src/pySODM/models/__init__.py b/src/pySODM/models/__init__.py
@@ -0,0 +1,17 @@
+"""
+By setting the following environment variables, pySODM prohibits Numpy, Torch & Tensorflow from automatically multithreading large matrix operations
+This does not play nice with the pySODM feature of running the model N times in parallel using multiprocessing
+"""
+
+import os
+
+# Numpy & Torch
+os.environ['VECLIB_MAXIMUM_THREADS'] = '1'
+os.environ['OPENBLAS_NUM_THREADS'] = '1'
+os.environ['MKL_NUM_THREADS'] = '1'
+os.environ['BLIS_NUM_THREADS'] = '1'
+os.environ['NUMEXPR_NUM_THREADS'] = '1'
+
+# Tensorflow
+os.environ['TF_NUM_INTEROP_THREADS'] = '1'
+os.environ['TF_NUM_INTRAOP_THREADS'] = '1'