Pin numpy<2.0.0

.github/workflows/integration-test-slow.yml

@@ -147,15 +147,9 @@ jobs:
         run: |
           pip install uv
           uv pip install --system "pyyaml==5.3.1" docker-compose
-      - name: Install Linux System Dependencies
-        if: (inputs.os == 'ubuntu-latest' || inputs.os == 'arc-runner-set')
-        run: sudo apt install graphviz
       - name: Install MacOS System Dependencies
         if: runner.os=='macOS'
-        run: brew install graphviz libomp
-      - name: Install Windows System Dependencies
-        if: runner.os=='Windows'
-        run: choco install graphviz
+        run: brew install libomp
       - name: Unbreak Python in GHA for 3.8-3.10
         if: runner.os=='macOS' && inputs.python-version != '3.11'
         # github actions overwrites brew's python. Force it to reassert itself, by

pyproject.toml

@@ -50,6 +50,7 @@ distro = "^1.6.0"
 docker = "~6.1.0"
 httplib2 = "<0.20,>=0.19.1"
 gitpython = "^3.1.18"
+numpy = "<2.0.0"
 pandas = ">=1.1.5"
 passlib = { extras = ["bcrypt"], version = "~1.7.4" }
 psutil = ">=5.0.0"
@@ -417,7 +418,6 @@ module = [
     "IPython.core.*",
     "IPython.display.*",
     "plotly.*",
-    "graphviz.*",
     "dash.*",
     "dash_bootstrap_components.*",
     "dash_cytoscape",

scripts/test-migrations.sh

@@ -211,7 +211,7 @@ function test_upgrade_to_version() {
     else
         uv pip install "zenml[templates,server]==$VERSION"
         # handles unpinned sqlmodel dependency in older versions
-        uv pip install "sqlmodel==0.0.8" "bcrypt==4.0.1" "pyyaml-include<2.0"
+        uv pip install "sqlmodel==0.0.8" "bcrypt==4.0.1" "pyyaml-include<2.0" "numpy<2.0.0" "tenacity!=8.4.0"
     fi
 
     # Get the major and minor version of Python

src/zenml/integrations/airflow/__init__.py

@@ -32,7 +32,11 @@ class AirflowIntegration(Integration):
     NAME = AIRFLOW
     # remove pendulum version requirement once Airflow supports
     # pendulum>-3.0.0
-    REQUIREMENTS = ["apache-airflow~=2.4.0", "pendulum<3.0.0"]
+    REQUIREMENTS = [
+        "apache-airflow~=2.4.0",
+        "pendulum<3.0.0",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def flavors(cls) -> List[Type[Flavor]]:

src/zenml/integrations/constants.py

@@ -30,7 +30,6 @@
 GCP = "gcp"
 GITHUB = "github"
 GITLAB = "gitlab"
-GRAPHVIZ = "graphviz"
 HUGGINGFACE = "huggingface"
 HYPERAI = "hyperai"
 GREAT_EXPECTATIONS = "great_expectations"

src/zenml/integrations/deepchecks/__init__.py

@@ -41,6 +41,7 @@ class DeepchecksIntegration(Integration):
         "pandas<2.0.0",
         "opencv-python==4.5.5.64",  # pin to same version
         "opencv-python-headless==4.5.5.64",  # pin to same version
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
     ]
     APT_PACKAGES = ["ffmpeg", "libsm6", "libxext6"]
 

src/zenml/integrations/evidently/__init__.py

@@ -54,7 +54,10 @@ class EvidentlyIntegration(Integration):
     """[Evidently](https://github.com/evidentlyai/evidently) integration for ZenML."""
 
     NAME = EVIDENTLY
-    REQUIREMENTS = ["evidently>0.2.6,<0.4.5"]  # supports pyyaml 6
+    REQUIREMENTS = [
+        "evidently>0.2.6,<0.4.5",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]  # supports pyyaml 6
 
     @classmethod
     def flavors(cls) -> List[Type[Flavor]]:

src/zenml/integrations/langchain/__init__.py

@@ -25,7 +25,11 @@ class LangchainIntegration(Integration):
     """Definition of langchain integration for ZenML."""
 
     NAME = LANGCHAIN
-    REQUIREMENTS = ["langchain==0.0.325", "pyyaml>=6.0.1"]
+    REQUIREMENTS = [
+        "langchain==0.0.325",
+        "pyyaml>=6.0.1",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def activate(cls) -> None:

src/zenml/integrations/neural_prophet/__init__.py

@@ -21,7 +21,11 @@ class NeuralProphetIntegration(Integration):
     """Definition of NeuralProphet integration for ZenML."""
 
     NAME = NEURAL_PROPHET
-    REQUIREMENTS = ["neuralprophet>=0.3.2,<0.5.0", "holidays>=0.4.1,<0.25.0"]
+    REQUIREMENTS = [
+        "neuralprophet>=0.3.2,<0.5.0",
+        "holidays>=0.4.1,<0.25.0",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def activate(cls) -> None: