From 15c4478e2feab9b42065ae98825dca08c08fdcd8 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]"
 <41898282+github-actions[bot]@users.noreply.github.com>
Date: Tue, 20 Aug 2024 09:01:43 +0200
Subject: [PATCH] Support PyTorch 2.4.0 (#1611)

* Support latest PyTorch release

* Update bug_report.yml

* Update bug_report.yml

* Update ci.yaml

* Update setup.py

---------

Co-authored-by: mtar <18899420+mtar@users.noreply.github.com>
Co-authored-by: Michael Tarnawa <m.tarnawa@fz-juelich.de>
Co-authored-by: Claudia Comito <39374113+ClaudiaComito@users.noreply.github.com>
---
 .github/ISSUE_TEMPLATE/bug_report.yml | 12 +++++-------
 .github/workflows/ci.yaml             |  1 +
 setup.py                              |  4 ++--
 3 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/.github/ISSUE_TEMPLATE/bug_report.yml b/.github/ISSUE_TEMPLATE/bug_report.yml
index e73f744681..febdbbade2 100644
--- a/.github/ISSUE_TEMPLATE/bug_report.yml
+++ b/.github/ISSUE_TEMPLATE/bug_report.yml
@@ -34,8 +34,8 @@ body:
       description: What version of Heat are you running?
       options:
         - main (development branch)
+        - 1.4.x
         - 1.3.x
-        - 1.2.x
     validations:
       required: true
   - type: dropdown
@@ -44,23 +44,21 @@ body:
       label: Python version
       description: What Python version?
       options:
+        - 3.12
         - 3.11
         - "3.10"
         - 3.9
-        - 3.8
   - type: dropdown
     id: pytorch-version
     attributes:
       label: PyTorch version
       description: What PyTorch version?
       options:
+        - 2.4
+        - 2.3
         - 2.2
         - 2.1
-        - 2.0
-        - 1.13
-        - 1.12
-        - 1.11
-        - "1.10"
+        - '2.0'
   - type: textarea
     id: mpi-version
     attributes:
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index f7a980fd03..daec3fac63 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -23,6 +23,7 @@ jobs:
           - 'torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2'
           - '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.0 torchvision==0.19.0 torchaudio==2.4.0'
         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 b98401ec17..78931ab36b 100644
--- a/setup.py
+++ b/setup.py
@@ -35,10 +35,10 @@
     install_requires=[
         "mpi4py>=3.0.0, <4.0.0",
         "numpy>=1.22.0, <2",
-        "torch>=2.0.0, <2.3.2",
+        "torch>=2.0.0, <2.4.1",
         "scipy>=1.10.0",
         "pillow>=6.0.0",
-        "torchvision>=0.15.2",
+        "torchvision>=0.15.2, <0.19.1",
     ],
     extras_require={
         "docutils": ["docutils>=0.16"],