diff --git a/executorlib/interactive/slurm.py b/executorlib/interactive/slurm.py index 8c962529..7ed465d1 100644 --- a/executorlib/interactive/slurm.py +++ b/executorlib/interactive/slurm.py @@ -7,7 +7,9 @@ def validate_max_workers(max_workers: int, cores: int, threads_per_core: int): - cores_total = int(os.environ["SLURM_NTASKS"]) * int(os.environ["SLURM_CPUS_PER_TASK"]) + cores_total = int(os.environ["SLURM_NTASKS"]) * int( + os.environ["SLURM_CPUS_PER_TASK"] + ) cores_requested = max_workers * cores * threads_per_core if cores_total < cores_requested: raise ValueError(