Supply default value for worker_procs vatlab/sos-notebook#237

vatlab · Aug 8, 2019 · 4b6fb6e · 4b6fb6e
1 parent 655ef11
commit 4b6fb6e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/sos/workers.py b/src/sos/workers.py
@@ -390,6 +390,9 @@ class WorkerManager(object):
     def __init__(self, worker_procs, backend_socket):
         if isinstance(worker_procs, (int, str)):
             self._worker_procs = [str(worker_procs)]
+        elif worker_procs is None:
+            # from sos notebook with no value specified
+            self._worker_procs = [str(min(max(os.cpu_count() // 2, 2), 8))]
         else:
             # should be a sequence
             self._worker_procs = worker_procs