Merge pull request #20 from zenml-io/bug/fix-sklearn-vulnerability

Fix scikit-learn vulnerability

requirements.txt

@@ -1,4 +1,4 @@
-scikit-learn<1.3
+scikit-learn
 copier
 jinja2-time
 zenml[server]>=0.52.0

template/requirements.txt

@@ -1,4 +1,4 @@
 zenml[server]>=0.50.0
 notebook
-scikit-learn<1.3
+scikit-learn
 pyarrow

tests/conftest.py

@@ -13,7 +13,9 @@
 #  permissions and limitations under the License.
 
 
+import contextlib 
 import os
+import sys
 import shutil
 from typing import Generator
 
@@ -97,4 +99,8 @@ def clean_zenml_client(
 
     # remove all traces, and change working directory back to base path
     os.chdir(orig_cwd)
-    shutil.rmtree(str(tmp_path))
+    if sys.platform == "win32":
+        with contextlib.suppress(Exception):  
+            shutil.rmtree(str(tmp_path))  
+    else:
+        shutil.rmtree(str(tmp_path))