foo

Author: TheMichaelHu

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()