Skip to content

Commit 266b9f1

Browse files
committed
mypy and ruff fixes
1 parent c0093c3 commit 266b9f1

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

vec_inf/cli/_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,7 +321,7 @@ def batch_launch(
321321
is_flag=True,
322322
help="Output in JSON string",
323323
)
324-
def status(slurm_job_id: str = None, json_mode: bool = False) -> None:
324+
def status(slurm_job_id: Optional[str] = None, json_mode: bool = False) -> None:
325325
"""Get the status of a running model on the cluster.
326326
327327
Parameters

vec_inf/client/api.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,6 @@ def fetch_running_jobs(self) -> list[str]:
193193
List of matching job names; empty list if squeue unavailable.
194194
"""
195195
try:
196-
# Run squeue for current user
197196
res = subprocess.run(
198197
["squeue", "--me", "--noheader"],
199198
capture_output=True,
@@ -217,7 +216,6 @@ def fetch_running_jobs(self) -> list[str]:
217216
text=True,
218217
check=True,
219218
)
220-
# Example: "JobId=12345 JobName=my-long-job-name-vec-inf ..."
221219
m = re.search(r"\bJobName=([^\s]+)", sctl.stdout)
222220
if m and m.group(1).endswith("-vec-inf"):
223221
matching_ids.append(jid)

0 commit comments

Comments
 (0)