diff --git a/requirements.txt b/requirements.txt index 20c5679..c15add1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,4 @@ joblib==1.3.2 pandas==2.0.3 pytest==7.4.0 scikit-learn==1.3.0 -torch==2.0.1 \ No newline at end of file +torch==2.2.0 \ No newline at end of file diff --git a/setup.py b/setup.py index 15450ca..4f4f4c5 100644 --- a/setup.py +++ b/setup.py @@ -99,7 +99,7 @@ def run_tests(self): # your project is installed. For an analysis of "install_requires" vs pip's # requirements files see: # https://packaging.python.org/en/latest/requirements.html - install_requires=["torch==2.0.1", "pandas==2.0.3", + install_requires=["torch==2.2.0", "pandas==2.0.3", "joblib==1.3.2", "scikit-learn==1.3.0",], # List additional groups of dependencies here (e.g. development # dependencies). You can install these using the following syntax,