Skip to content

Commit

Permalink
If you want something done right, do it yourself
Browse files Browse the repository at this point in the history
  • Loading branch information
ProbablyFaiz committed Aug 25, 2024
1 parent 87d204c commit 284598a
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion rl/cli/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -359,9 +359,11 @@ def _get_all_jobs(partition: str | None = None, show_progress=False):
progress.update(task, completed=1)
return results

squeue_command = ["squeue", "-u", CURRENT_USER, "-h", "-o", "%A %j %u %P %N %t %M %L"]
squeue_command = ["squeue", "-h", "-o", "%A %j %u %P %N %t %M %L"]
if partition:
squeue_command.extend(["-p", partition])
else:
squeue_command.extend(["-u", CURRENT_USER])

output = subprocess.run(
squeue_command,
Expand Down

0 comments on commit 284598a

Please sign in to comment.