Fix Windows subprocess issue

Author: Alexsandruss

sklbench/utils/env.py

@@ -47,9 +47,15 @@ def get_numa_cpus_conf() -> Dict[int, str]:
 
 def get_number_of_sockets():
     if sys.platform == "win32":
-        command = "wmic cpu get DeviceID"
-        result = subprocess.check_output(command, shell=False, text=True)
-        n_sockets = len(list(filter(lambda x: x.startswith("CPU"), result.split("\n"))))
+        try:
+            command = ["wmic", "cpu", "get", "DeviceID"]
+            result = subprocess.check_output(command, shell=False, text=True)
+            n_sockets = len(
+                list(filter(lambda x: x.startswith("CPU"), result.split("\n")))
+            )
+        except (FileNotFoundError, subprocess.CalledProcessError, ValueError, IndexError):
+            logger.warning("Unable to get number of sockets via wmic")
+            n_sockets = 1
     elif sys.platform == "linux":
         try:
             _, lscpu_text, _ = read_output_from_command("lscpu")

sklbench/utils/measurement.py

@@ -124,7 +124,9 @@ def get_n_from_cache_size():
     return ceil(sqrt(n_sockets * cache_size / 8))
 
 
-def flush_cache(n: int = get_n_from_cache_size()):
+def flush_cache(n: int | None = None):
+    if n is None:
+        n = get_n_from_cache_size()
     np.matmul(np.random.rand(n, n), np.random.rand(n, n))