fixing the CI awaiting the mlstacks release (#2792)

zenml-io · Jun 25, 2024 · d5ba2fe · d5ba2fe
1 parent 1de7adf
commit d5ba2fe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/.github/workflows/ci-slow.yml b/.github/workflows/ci-slow.yml
@@ -142,12 +142,11 @@ jobs:
       - name: Create virtual environment
         run: |
           uv venv
-#      TODO: REACTIVATE BEFORE THE RELEASE
-#      - name: Check mlstacks compatibility
-#        run: |
-#          source .venv/bin/activate
-#          uv pip install -e .
-#          uv pip install mlstacks
+      - name: Check mlstacks compatibility
+        run: |
+          source .venv/bin/activate
+          uv pip install -e .
+          uv pip install mlstacks
       - name: Check for broken dependencies
         run: |
           source .venv/bin/activate

diff --git a/scripts/install-zenml-dev.sh b/scripts/install-zenml-dev.sh
@@ -28,10 +28,7 @@ parse_args () {
 
 install_zenml() {
     # install ZenML in editable mode
-    uv pip install $PIP_ARGS -e ".[server,templates,terraform,secrets-aws,secrets-gcp,secrets-azure,secrets-hashicorp,s3fs,gcsfs,adlfs,dev]"
-
-    # TODO: Remove. Temporary installation from the upgrade branch.
-    uv pip install $PIP_ARGS git+https://github.com/zenml-io/mlstacks.git@develop
+    uv pip install $PIP_ARGS -e ".[server,templates,terraform,secrets-aws,secrets-gcp,secrets-azure,secrets-hashicorp,s3fs,gcsfs,adlfs,dev,mlstacks]"
 }
 
 install_integrations() {