Skip to content

Commit 770a6fc

Browse files
authored
Update test_status_callbacks.py
Added some type annotations
1 parent 5aac32c commit 770a6fc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/getting_started/test_status_callbacks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import time
2-
from psij import Job, JobSpec, JobExecutor
2+
from psij import Job, JobSpec, JobStatus, JobExecutor
33
from executor_test_params import ExecutorTestParams
44

55
count = 10
66

77

88
def test_getting_started_scheduling_info(execparams: ExecutorTestParams) -> None:
9-
def callback(job, status):
9+
def callback(job: Job, status: JobStatus) -> None:
1010
global count
1111

1212
if status.final:

0 commit comments

Comments
 (0)