diff --git a/template/gradio/requirements.txt b/template/gradio/requirements.txt index 4ae46ad..1bddfdf 100644 --- 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 \ No newline at end of file diff --git a/template/requirements.txt b/template/requirements.txt index ee4e7af..f7f9817 100644 --- 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 index 5ec3260..d2be666 100644 --- 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))