diff --git a/tests/system/aiplatform/test_e2e_forecasting.py b/tests/system/aiplatform/test_e2e_forecasting.py index 76937671226..6f70578d8f7 100644 --- a/tests/system/aiplatform/test_e2e_forecasting.py +++ b/tests/system/aiplatform/test_e2e_forecasting.py @@ -28,7 +28,6 @@ "bq://ucaip-sample-tests:ucaip_test_us_central1.2021_sales_predict" ) - @pytest.mark.usefixtures("prepare_staging_bucket", "delete_staging_bucket") class TestEndToEndForecasting(e2e_base.TestEndToEnd): """End to end system test of the Vertex SDK with forecasting data.""" @@ -151,4 +150,5 @@ def test_end_to_end_forecasting(self, shared_state): ) finally: for resource in resources: + resource.wait_for_resource_creation() resource.delete()