feat: Remove deletion of model upload staging bucket in system tests

matthew29tang · copybara-github · commit b9b373b69eab · 2024-01-29T16:22:20.000-08:00
PiperOrigin-RevId: 602531376
diff --git a/tests/system/aiplatform/test_model_upload.py b/tests/system/aiplatform/test_model_upload.py
@@ -28,7 +28,7 @@
 _XGBOOST_MODEL_URI = "gs://cloud-samples-data-us-central1/vertex-ai/google-cloud-aiplatform-ci-artifacts/models/iris_xgboost/model.bst"
 
 
-@pytest.mark.usefixtures("delete_staging_bucket", "tear_down_resources")
+@pytest.mark.usefixtures("tear_down_resources")
 class TestModelUploadAndUpdate(e2e_base.TestEndToEnd):
 
     _temp_prefix = "temp_vertex_sdk_e2e_model_upload_test"
@@ -59,8 +59,6 @@ def test_upload_and_deploy_xgboost_model(self, shared_state):
         # Checking that the bucket is auto-generated
         assert "-vertex-staging-" in staging_bucket.name
 
-        shared_state["bucket"] = staging_bucket
-
         # Currently we need to explicitly specify machine type.
         # See https://github.com/googleapis/python-aiplatform/issues/773
         endpoint = model.deploy(machine_type="n1-standard-2")
diff --git a/tests/system/aiplatform/test_model_version_management.py b/tests/system/aiplatform/test_model_version_management.py
@@ -28,7 +28,7 @@
 from tests.system.aiplatform import test_model_upload
 
 
-@pytest.mark.usefixtures("delete_staging_bucket", "tear_down_resources")
+@pytest.mark.usefixtures("tear_down_resources")
 class TestVersionManagement(e2e_base.TestEndToEnd):
 
     _temp_prefix = "temp_vertex_sdk_e2e_model_upload_test"
@@ -66,8 +66,6 @@ def test_upload_deploy_manage_versioned_model(self, shared_state):
         # Checking that the bucket is auto-generated
         assert "-vertex-staging-" in staging_bucket.name
 
-        shared_state["bucket"] = staging_bucket
-
         assert model.version_description == version_description
         assert model.version_aliases == version_aliases
         assert "default" in model.version_aliases
