Fix PyYAML and linter errors

zenml-io · Nov 22, 2023 · 1d7c33f · 1d7c33f
1 parent 4065cf5
commit 1d7c33f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -349,6 +349,7 @@ module = [
     "datasets.*",
     "pyngrok.*",
     "cloudpickle.*",
+    "matplotlib.*",
 ]
 ignore_missing_imports = true
 

diff --git a/scripts/install-zenml-dev.sh b/scripts/install-zenml-dev.sh
@@ -48,11 +48,16 @@ install_integrations() {
     zenml integration export-requirements \
         --output-file integration-requirements.txt \
         $ignore_integrations_args
+
+    # pin pyyaml>=6.0.1
+    echo "" >> integration-requirements.txt
+    echo "pyyaml>=6.0.1" >> integration-requirements.txt
+
     pip install -r integration-requirements.txt
     rm integration-requirements.txt
 
-    # install langchain and llama_index integrations separately
-    zenml integration install -y langchain llama_index
+    # install langchain separately
+    zenml integration install -y langchain
 }
 
 

diff --git a/src/zenml/integrations/langchain/__init__.py b/src/zenml/integrations/langchain/__init__.py
@@ -25,7 +25,7 @@ class LangchainIntegration(Integration):
     """Definition of langchain integration for ZenML."""
 
     NAME = LANGCHAIN
-    REQUIREMENTS = ["langchain>=0.0.116"]
+    REQUIREMENTS = ["langchain>=0.0.116", "pyyaml>=6.0.1"]
 
     @classmethod
     def activate(cls) -> None: