Skip to content

Commit b4ec000

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: migration to python 3.12. called_once was deleted in favor of assert_called_once, since it was largely misused
PiperOrigin-RevId: 733699828
1 parent 26efb7b commit b4ec000

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

tests/unit/aiplatform/test_models.py

+3-9
Original file line numberDiff line numberDiff line change
@@ -4240,9 +4240,7 @@ def test_model_evaluate_with_gcs_input_uris(
42404240

42414241
assert isinstance(eval_job, model_evaluation_job._ModelEvaluationJob)
42424242

4243-
assert mock_pipeline_service_create.called_once
4244-
4245-
assert mock_pipeline_service_get.called_once
4243+
mock_pipeline_service_create.assert_called_once()
42464244

42474245
eval_job.wait()
42484246

@@ -4283,9 +4281,7 @@ def test_model_evaluate_with_bigquery_input(
42834281

42844282
assert isinstance(eval_job, model_evaluation_job._ModelEvaluationJob)
42854283

4286-
assert mock_pipeline_service_create.called_once
4287-
4288-
assert mock_pipeline_service_get.called_once
4284+
mock_pipeline_service_create.assert_called_once()
42894285

42904286
@pytest.mark.parametrize(
42914287
"job_spec_json",
@@ -4315,9 +4311,7 @@ def test_model_evaluate_using_initialized_staging_bucket(
43154311

43164312
assert isinstance(eval_job, model_evaluation_job._ModelEvaluationJob)
43174313

4318-
assert mock_pipeline_service_create.called_once
4319-
4320-
assert mock_pipeline_service_get.called_once
4314+
mock_pipeline_service_create.assert_called_once()
43214315

43224316
def test_model_evaluate_with_no_staging_path_or_initialized_staging_bucket_raises(
43234317
self,

0 commit comments

Comments
 (0)