Count CPUs to set # of tabs

Author: ayjayt

src/py/kaleido/_mocker.py

@@ -2,6 +2,7 @@
 
 import argparse
 import asyncio
+import multiprocessing
 import sys
 import time
 import warnings
@@ -16,6 +17,8 @@
 
 _logger = logistro.getLogger(__name__)
 
+cpus = multiprocessing.cpu_count()
+
 # Extract jsons of mocks
 test_dir = Path(__file__).resolve().parent.parent / "integration_tests"
 in_dir = test_dir / "mocks"
@@ -80,7 +83,12 @@ def _load_figures_from_paths(paths: list[Path]):
     help="Set the logging level (default INFO)",
 )
 
-parser.add_argument("--n", type=int, default=4, help="Number of tabs, default 4")
+parser.add_argument(
+    "--n",
+    type=int,
+    default=cpus,
+    help="Number of tabs, defaults to # of cpus",
+)
 
 parser.add_argument(
     "--input",