move aiplatform install to correct test suite

GoogleCloudPlatform · Jun 17, 2024 · 2c837af · 2c837af
1 parent 4f34ba3
commit 2c837af
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/.github/workflows/.github-ci.yml b/.github/workflows/.github-ci.yml
@@ -45,7 +45,6 @@ jobs:
         run: |
           python3 -m pip install --upgrade pip
           pip3 install -r requirements.txt
-          pip3 install google-cloud-aiplatform
       - name: Run unit tests
         run: |
           pytest tests/unit --cov=AutoMLOps=
@@ -66,6 +65,7 @@ jobs:
         run: |
           python3 -m pip install --upgrade pip
           pip3 install -r requirements.txt
+          pip3 install google-cloud-aiplatform
       - name: 'Authenticate to Google Cloud'
         uses: 'google-github-actions/auth@v2'
         with:

diff --git a/tests/integration/test_beans_training_model.py b/tests/integration/test_beans_training_model.py
@@ -200,17 +200,14 @@ def test_beans_training_model():
                     naming_prefix=model_id
     )
 
-
     AutoMLOps.provision(hide_warnings=False)
     time.sleep(300)
-    print('HIT PROVISION')
 
     # Assert that GCP infrastructure was stood up with the correct names.
     defaults = read_yaml_file(GENERATED_DEFAULTS_FILE)
     helpers.assert_successful_provisioning(defaults)
 
     # Assert that Vertex AI endpoint was created and returns predictions.
-
     from google.cloud import aiplatform
     aiplatform.init(project=project_id)
     endpoints = aiplatform.Endpoint.list()