From b912846ce6a8599c911139d36c20211455d67aa4 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]"
 <41898282+github-actions[bot]@users.noreply.github.com>
Date: Thu, 31 Oct 2024 09:44:29 +0100
Subject: [PATCH] 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 <18899420+mtar@users.noreply.github.com>
Co-authored-by: Fabian Hoppe <112093564+mrfh92@users.noreply.github.com>
---
 .github/ISSUE_TEMPLATE/bug_report.yml | 2 +-
 .github/workflows/ci.yaml             | 1 +
 setup.py                              | 4 ++--
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml
index e4381f63d..5ef72f6bd 100644
--- a/.github/ISSUE_TEMPLATE/bug_report.yml
+++ b/.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
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 4884c1eea..6309d4d68 100644
--- a/.github/workflows/ci.yaml
+++ b/.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'
diff --git a/setup.py b/setup.py
index 52190877e..bb71bf1c3 100644
--- a/setup.py
+++ b/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"],