Temporary fix for Tensorflow NUMEXPR_MAX_THREADS

See compomics/DeepLC#19
compomics · Feb 10, 2021 · 9d91e35 · 9d91e35
1 parent be028ec
commit 9d91e35
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ms2rescore/retention_time.py b/ms2rescore/retention_time.py
@@ -63,6 +63,10 @@ def __init__(
         self.calibration_set_size = calibration_set_size
         self.num_cpu = num_cpu
 
+        # Until fixed upstream: https://github.com/compomics/DeepLC/issues/19
+        if "NUMEXPR_MAX_THREADS" not in os.environ:
+            os.environ['NUMEXPR_MAX_THREADS'] = str(self.num_cpu)
+
         from deeplc import DeepLC
 
         self.deeplc_predictor = DeepLC(