From 33d5ade2eaddedd119696892588227d1467db0bc Mon Sep 17 00:00:00 2001 From: Ulada Zakharava Date: Wed, 22 May 2024 15:47:40 +0000 Subject: [PATCH] Fix MLEngine system test --- .../providers/google/cloud/ml_engine/example_mlengine.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/system/providers/google/cloud/ml_engine/example_mlengine.py b/tests/system/providers/google/cloud/ml_engine/example_mlengine.py index 6bed016d350d4..87602da88c46c 100644 --- a/tests/system/providers/google/cloud/ml_engine/example_mlengine.py +++ b/tests/system/providers/google/cloud/ml_engine/example_mlengine.py @@ -57,7 +57,7 @@ PROJECT_ID = os.environ.get("SYSTEM_TESTS_GCP_PROJECT", "default") ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID", "default") -DAG_ID = "example_gcp_mlengine" +DAG_ID = "gcp_mlengine" REGION = "us-central1" PACKAGE_DISPLAY_NAME = f"package-{DAG_ID}-{ENV_ID}".replace("_", "-") @@ -222,7 +222,7 @@ task_id="delete_model_version_1", project_id=PROJECT_ID, region=REGION, - model_id=model_id_v2, + model_id=model_id_v1, trigger_rule=TriggerRule.ALL_DONE, ) # [END howto_operator_gcp_mlengine_delete_version] @@ -232,7 +232,7 @@ task_id="delete_model", project_id=PROJECT_ID, region=REGION, - model_id=model_id_v1, + model_id=model_id_v2, trigger_rule=TriggerRule.ALL_DONE, ) # [END howto_operator_gcp_mlengine_delete_model]