Deal with case where sinfo (partitions) do not include the number o…

…f cores
basnijholt · Dec 1, 2023 · 882e51d · 882e51d
1 parent 2084e27
commit 882e51d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/adaptive_scheduler/_scheduler/slurm.py b/adaptive_scheduler/_scheduler/slurm.py
@@ -286,6 +286,8 @@ def partitions(self) -> dict[str, int]:
 
 def _get_ncores(partition: str) -> int:
     numbers = re.findall(r"\d+", partition)
+    if not numbers:
+        return None
     return int(numbers[0])
 
 
@@ -294,7 +296,7 @@ def slurm_partitions(
     *,
     timeout: int = 5,
     with_ncores: bool = True,
-) -> list[str] | dict[str, int]:
+) -> list[str] | dict[str, int | None]:
     """Get the available slurm partitions, raises subprocess.TimeoutExpired after timeout."""
     output = subprocess.run(
         ["sinfo", "-ahO", "partition"],