Control thread pool with EXTRA_NUM_THREADS environment variable

Author: takluyver

extra_data/utils.py

@@ -27,7 +27,7 @@ def default_num_threads(fixed_limit=16):
     # Default to 16, OMP_NUM_THREADS, or available CPU cores (picking lowest)
     threads_limits = ([fixed_limit, available_cpu_cores()])
     try:
-        threads_limits.append(int(os.environ['OMP_NUM_THREADS']))
+        threads_limits.append(int(os.environ['EXTRA_NUM_THREADS']))
     except (KeyError, ValueError):  # Not set, or not an integer
         pass
     return min(threads_limits)