Skip to content

Commit a8a3ad8

Browse files
authored
fix: uses str job_id (#16)
1 parent b953790 commit a8a3ad8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/aind_airflow_jobs/slurm_handler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@ def _monitor_job(
386386

387387
job_id = submit_response.job_id
388388
job_name = self.job_properties.name
389-
job_response = self.slurm.slurm_v0040_get_job(job_id=job_id)
389+
job_response = self.slurm.slurm_v0040_get_job(job_id=str(job_id))
390390
errors = job_response.errors
391391
start_time = (
392392
None if not job_response.jobs else job_response.jobs[0].start_time
@@ -411,7 +411,7 @@ def _monitor_job(
411411
)
412412
while not is_finished and not errors:
413413
sleep(self.polling_request_sleep)
414-
job_response = self.slurm.slurm_v0040_get_job(job_id=job_id)
414+
job_response = self.slurm.slurm_v0040_get_job(job_id=str(job_id))
415415
errors = job_response.errors
416416
start_time = (
417417
None

0 commit comments

Comments
 (0)