diff --git a/optional-requirements.txt b/optional-requirements.txt index 954908a1..c59a48a5 100644 --- a/optional-requirements.txt +++ b/optional-requirements.txt @@ -1,3 +1,3 @@ scikit-mdr==0.4.4 skrebate==0.3.4 -torch==1.13.1 +torch==2.2.0 diff --git a/setup.py b/setup.py index 695ab8b1..18ac5b45 100644 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ def calculate_version(): 'dask': ['dask>=0.18.2', 'distributed>=1.22.1', 'dask-ml>=1.0.0'], - 'torch': ['torch==1.13.1'], + 'torch': ['torch==2.2.0'], 'imblearn': ['imbalanced-learn>=0.7.0'] }, classifiers=[