threading library as dependency

photochem/__init__.py

@@ -1,7 +1,8 @@
-import os
-os.environ['OMP_NUM_THREADS'] = '1'
-
 from ._photochem import EvoAtmosphere, Atmosphere, PhotoException, __version__
 from photochem_clima_data import DATA_DIR
 
+# Limits OpenMP threads to 1
+from threadpoolctl import threadpool_limits
+_ = threadpool_limits(limits=1, user_api='openmp')
+
 zahnle_earth = DATA_DIR+'/reaction_mechanisms/zahnle_earth.yaml'

photochem/clima.py

@@ -1,6 +1,7 @@
-import os
-os.environ['OMP_NUM_THREADS'] = '1'
-
 from ._clima import AdiabatClimate, ClimaException
 from ._clima import rebin # rebin routine from futils
-from ._clima import __version__
+from ._clima import __version__
+
+# Limits OpenMP threads to 1
+from threadpoolctl import threadpool_limits
+_ = threadpool_limits(limits=1, user_api='openmp')

setup.py

@@ -18,7 +18,7 @@
     python_requires='>=3.6',
     version=version,
     license="GNU General Public License v3.0",
-    install_requires=['numpy','scipy','pyyaml', 'numba', 'h5py', 'astropy'], 
+    install_requires=['numpy','scipy','pyyaml', 'numba', 'h5py', 'astropy', 'threadpoolctl'], 
     author='Nicholas Wogan',
     author_email = '[email protected]',
     description = "Photochemical model of planet's atmospheres.",