File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
tests/sap/aibus/dar/client Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -464,9 +464,9 @@ def test_is_job_failed(self):
464
464
non_failed_job = self ._make_job_resource (non_failed_state )
465
465
assert ModelManagerClient .is_job_failed (non_failed_job ) is False
466
466
467
- def test_read_job_by_model_name (self ):
468
- job1 = self ._make_job_resource ()
469
- job2 = self ._make_job_resource ()
467
+ def test_read_job_by_model_name (self , model_manager_client : ModelManagerClient ):
468
+ job1 = self ._make_job_resource ("SUCCEEDED" )
469
+ job2 = self ._make_job_resource ("SUCCEEDED" )
470
470
job2 ["modelName" ] = "my-model-2"
471
471
job_collection_response = {"count" : 2 , "jobs" : [job1 , job2 ]}
472
472
model_manager_client .read_job_collection = create_autospec (
@@ -477,7 +477,9 @@ def test_read_job_by_model_name(self):
477
477
assert response ["modelName" ] == "my-model-1"
478
478
model_manager_client .read_job_collection .assert_called_once ()
479
479
480
- def test_read_job_by_model_name_job_not_found (self ):
480
+ def test_read_job_by_model_name_job_not_found (
481
+ self , model_manager_client : ModelManagerClient
482
+ ):
481
483
job1 = self ._make_job_resource ("SUCCEEDED" )
482
484
job2 = self ._make_job_resource ("SUCCEEDED" )
483
485
job2 ["modelName" ] = "my-model-2"
You can’t perform that action at this time.
0 commit comments