Support PyTorch 2.5.1 (#1701)

* Support latest PyTorch release

* Update dependencies

* Update bug_report.yml

* Update ci.yaml

* Update setup.py

---------

Co-authored-by: mtar <[email protected]>
Co-authored-by: Fabian Hoppe <[email protected]>

.github/ISSUE_TEMPLATE/bug_report.yml

@@ -35,7 +35,6 @@ body:
       options:
         - main (development branch)
         - 1.5.x
-        - 1.4.x
     validations:
       required: true
   - type: dropdown
@@ -54,6 +53,7 @@ body:
       label: PyTorch version
       description: What PyTorch version?
       options:
+        - 2.5
         - 2.4
         - 2.3
         - 2.2

.github/workflows/ci.yaml

@@ -24,6 +24,7 @@ jobs:
           - 'torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2'
           - 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
           - 'torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1'
+          - 'torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1'
         exclude:
           - py-version: '3.12'
             pytorch-version: 'torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2'

setup.py

@@ -35,10 +35,10 @@
     install_requires=[
         "mpi4py>=3.0.0",
         "numpy>=1.22.0, <2",
-        "torch>=2.0.0, <2.4.2",
+        "torch>=2.0.0, <2.5.2",
         "scipy>=1.10.0",
         "pillow>=6.0.0",
-        "torchvision>=0.15.2, <0.19.2",
+        "torchvision>=0.15.2, <0.20.2",
     ],
     extras_require={
         "docutils": ["docutils>=0.16"],