fix: make new env var EDSNLP_MAX_CPU_WORKERS is optional

aphp · May 21, 2024 · b6cf1da · b6cf1da
1 parent d953be1
commit b6cf1da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/edsnlp/processing/multiprocessing.py b/edsnlp/processing/multiprocessing.py
@@ -796,7 +796,7 @@ def execute_multiprocessing_backend(
         and num_gpu_workers > 0
     )
 
-    num_cpus = int(os.environ["EDSNLP_MAX_CPU_WORKERS"] or cpu_count())
+    num_cpus = int(os.environ.get("EDSNLP_MAX_CPU_WORKERS") or cpu_count())
     num_devices = 0
     if requires_gpu:
         import torch