add manual aiplatform install

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

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/.github-ci.yml b/.github/workflows/.github-ci.yml
@@ -45,6 +45,7 @@ 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=

diff --git a/tests/integration/test_beans_training_model.py b/tests/integration/test_beans_training_model.py
@@ -181,8 +181,6 @@ def pipeline(bq_table: str,
 
 
 def test_beans_training_model():
-    # Importing here to avoid conflict with repeated import in deploy_model
-
     project_id = os.environ.get('PROJECT_ID')
     model_id = 'dry-beans-dt'
     ## Define the Pipeline Arguments