getting the lock state should look for billing as well

pitt-crc · Feb 28, 2024 · e3dae8c · e3dae8c
1 parent a016f80
commit e3dae8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bank/system/slurm.py b/bank/system/slurm.py
@@ -128,7 +128,7 @@ def get_locked_state(self, cluster: str) -> bool:
             raise ClusterNotFoundError(f'Cluster {cluster} is not configured with Slurm')
 
         cmd = f'sacctmgr -n -P show assoc account={self.account_name} format=GrpTresRunMins clusters={cluster}'
-        return 'cpu=0' in ShellCmd(cmd).out
+        return 'billing=0' in ShellCmd(cmd).out
 
     def set_locked_state(self, lock_state: bool, cluster: str) -> None:
         """Lock or unlock the current slurm account