diff --git a/optional-requirements.txt b/optional-requirements.txt index d71ccb2d..954908a1 100644 --- a/optional-requirements.txt +++ b/optional-requirements.txt @@ -1,3 +1,3 @@ scikit-mdr==0.4.4 skrebate==0.3.4 -torch==1.3.1 +torch==1.13.1 diff --git a/setup.py b/setup.py index ba37d5f2..8e802cc5 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.3.1'], + 'torch': ['torch==1.13.1'], 'imblearn': ['imbalanced-learn>=0.7.0'] }, classifiers=[