diff --git a/src/idefix_cli/_commands/run.py b/src/idefix_cli/_commands/run.py index fc9e63a..c6c0845 100644 --- 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