RFC: flatten logic tree

neutrinoceros · Nov 25, 2024 · d6970ec · d6970ec
1 parent 6505bdf
commit d6970ec
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/idefix_cli/_commands/run.py b/src/idefix_cli/_commands/run.py
@@ -146,13 +146,12 @@ def get_cpu_count() -> int:
     base_cpu_count: int | None
     if sys.version_info >= (3, 13):
         base_cpu_count = os.process_cpu_count()
+    elif hasattr(os, "sched_getaffinity"):
+        # this function isn't available on all platforms
+        base_cpu_count = len(os.sched_getaffinity(0))
     else:
-        if hasattr(os, "sched_getaffinity"):
-            # this function isn't available on all platforms
-            base_cpu_count = len(os.sched_getaffinity(0))
-        else:
-            # this proxy is good enough in most situations
-            base_cpu_count = os.cpu_count()
+        # this proxy is good enough in most situations
+        base_cpu_count = os.cpu_count()
     return base_cpu_count or 1