From cf2944e09217554cf44f79e73648a36e1da94644 Mon Sep 17 00:00:00 2001 From: Rosie Zou Date: Tue, 21 Jun 2022 17:22:41 -0700 Subject: [PATCH] fixing test bug --- tests/system/aiplatform/test_model_monitoring.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/system/aiplatform/test_model_monitoring.py b/tests/system/aiplatform/test_model_monitoring.py index bb394c15ae..c7d6547588 100644 --- a/tests/system/aiplatform/test_model_monitoring.py +++ b/tests/system/aiplatform/test_model_monitoring.py @@ -267,7 +267,7 @@ def test_mdm_one_model_one_valid_config(self, shared_state): with pytest.raises(core_exceptions.NotFound): job.api_client.get_model_deployment_monitoring_job(name=job_resource) - def test_mdm_two_models_one_valid_config(self, shared_state): + def test_mdm_two_models_two_valid_configs(self, shared_state): temp_endpoint_with_two_models = self.temp_endpoint_with_two_models(shared_state) [deployed_model1, deployed_model2] = list( map(lambda x: x.id, temp_endpoint_with_two_models.list_models()) @@ -283,7 +283,7 @@ def test_mdm_two_models_one_valid_config(self, shared_state): monitor_interval=MONITOR_INTERVAL, schedule_config=schedule_config, alert_config=alert_config, - objective_configs=objective_config, + objective_configs=all_configs, timeout=3600, project=e2e_base._PROJECT, location=e2e_base._LOCATION,