fix: handle when options are not passed

risuorg · Oct 27, 2022 · ec847a1 · ec847a1
1 parent abe0246
commit ec847a1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/risuclient/shell.py b/risuclient/shell.py
@@ -667,7 +667,10 @@ def dorisu(
     os.environ["CITELLUS_TMP"] = "%s" % tempfile.mkdtemp()
 
     # Set pool for same processes as CPU cores
-    p = Pool(options.numproc)
+    if options is not None:
+        p = Pool(options.numproc)
+    else:
+        p = Pool(cpu_count())
 
     # We've save path, use it
     if savepath: