Merge pull request #21 from zenml-io/bug/fix-vulnerabilities

Fix vulnerabilities
zenml-io · Jul 22, 2024 · 64082d7 · 64082d7
2 parents b807ffd + 6d72c38
commit 64082d7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/template/gradio/requirements.txt b/template/gradio/requirements.txt
@@ -7,6 +7,6 @@ datasets==2.12.0
 numpy==1.22.4
 pandas==1.5.3
 session_info==1.0.0
-scikit-learn==1.2.2
+scikit-learn==1.5.0
 transformers==4.28.1
 IPython==7.34.0
diff --git a/template/requirements.txt b/template/requirements.txt
@@ -1,4 +1,4 @@
 torchvision
 accelerate
 gradio
-zenml[server]==0.56.3
+zenml[server]>=0.56.3
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 
 
+import contextlib
 import os
 import sys
 import shutil
@@ -101,9 +102,7 @@ def clean_zenml_client(
     # remove all traces, and change working directory back to base path
     os.chdir(orig_cwd)
     if sys.platform == "win32":
-        try:
+        with contextlib.suppress(Exception):  
             shutil.rmtree(str(tmp_path))
-        except:
-            pass
     else:
         shutil.rmtree(str(tmp_path))