reverting mlstacks installation after the install (#2980)

zenml-io · Sep 9, 2024 · a63827a · a63827a
1 parent 28f81b4
commit a63827a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/.github/workflows/ci-slow.yml b/.github/workflows/ci-slow.yml
@@ -164,8 +164,7 @@ jobs:
         run: |
           source .venv/bin/activate
           uv pip install -e .
-          # TODO: Remove and add mlstacks back in
-          uv pip install git+https://github.com/zenml-io/mlstacks.git@feature/upgrade-python-3.12
+          uv pip install mlstacks
       - name: Check for broken dependencies
         run: |
           source .venv/bin/activate

diff --git a/pyproject.toml b/pyproject.toml
@@ -173,7 +173,7 @@ types-psutil = { version = "^5.8.13", optional = true }
 types-passlib = { version = "^1.7.7", optional = true }
 
 # mlstacks dependencies
-mlstacks = { version = "0.9.0", optional = true }
+mlstacks = { version = "0.10.0", optional = true }
 
 [tool.poetry.extras]
 server = [

diff --git a/scripts/install-zenml-dev.sh b/scripts/install-zenml-dev.sh
@@ -28,9 +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 and add the extra back in
-    uv pip install $PIP_ARGS git+https://github.com/zenml-io/mlstacks.git@feature/upgrade-python-3.12
+    uv pip install $PIP_ARGS -e ".[server,templates,terraform,secrets-aws,secrets-gcp,secrets-azure,secrets-hashicorp,s3fs,gcsfs,adlfs,dev,mlstacks]"
 }
 
 install_integrations() {