diff --git a/requirements.txt b/requirements.txt index d6be752..85dae18 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ pandas>=1.3.0 numpy joblib==1.3.1 -torch==2.0.1 +torch==2.2.0 tqdm==4.66.3 scikit-learn==1.3.0 diff --git a/setup.py b/setup.py index bf06f11..73a0a09 100644 --- a/setup.py +++ b/setup.py @@ -102,7 +102,7 @@ def run_tests(self): # https://packaging.python.org/en/latest/requirements.html install_requires=[ "pandas>=1.3.0", - "torch==2.0.1", + "torch==2.2.0", "joblib==1.3.1", "tqdm==4.66.3", "scikit-learn==1.3.0",