fix is_running

dcfidalgo · Nov 15, 2023 · f42c7f1 · f42c7f1
1 parent b11651e
commit f42c7f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/slurm_sweeps/backend.py b/src/slurm_sweeps/backend.py
@@ -99,7 +99,7 @@ def is_running() -> bool:
         """Are we running inside an `sbatch` call?"""
         try:
             subprocess.check_output(["sinfo"])
-            assert hasattr(os.environ, "SLURM_NTASKS")
+            assert "SLURM_NTASKS" in os.environ
         except (FileNotFoundError, subprocess.CalledProcessError, AssertionError):
             return False
         else: