Skip to content

Commit cc4c83a

Browse files
authored
fix: updates parsing start time (#17)
1 parent 813988b commit cc4c83a

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/aind_airflow_jobs/slurm_handler.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -389,7 +389,9 @@ def _monitor_job(
389389
job_response = self.slurm.slurm_v0040_get_job(job_id=str(job_id))
390390
errors = job_response.errors
391391
start_time = (
392-
None if not job_response.jobs else job_response.jobs[0].start_time
392+
None
393+
if not job_response.jobs or job_response.jobs[0].start_time is None
394+
else job_response.jobs[0].start_time.number
393395
)
394396
job_states = (
395397
None
@@ -416,7 +418,8 @@ def _monitor_job(
416418
start_time = (
417419
None
418420
if not job_response.jobs
419-
else job_response.jobs[0].start_time
421+
or job_response.jobs[0].start_time is None
422+
else job_response.jobs[0].start_time.number
420423
)
421424
job_states = (
422425
None
@@ -428,7 +431,7 @@ def _monitor_job(
428431
"job_id": job_id,
429432
"job_name": job_name,
430433
"job_states": job_states,
431-
"start_time": start_time.number,
434+
"start_time": start_time,
432435
}
433436
)
434437
logging.info(message)

0 commit comments

Comments
 (0)