From 485b0303dbdcfdafc7bd0b777b2dadee4b73cc38 Mon Sep 17 00:00:00 2001
From: mtar <18899420+mtar@users.noreply.github.com>
Date: Mon, 10 Jun 2024 10:44:29 +0000
Subject: [PATCH 1/2] Support latest PyTorch release

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 7680d3ea6b..6aca05e38c 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@
     install_requires=[
         "mpi4py>=3.0.0",
         "numpy>=1.22.0",
-        "torch>=1.12.0, <2.3.1",
+        "torch>=1.12.0, <2.3.2",
         "scipy>=1.10.0",
         "pillow>=6.0.0",
         "torchvision>=0.12.0",

From 7e0e55e0d0583a861736f0ddf6b8a23cb5aadd21 Mon Sep 17 00:00:00 2001
From: Michael Tarnawa <m.tarnawa@fz-juelich.de>
Date: Mon, 10 Jun 2024 12:59:42 +0200
Subject: [PATCH 2/2] Update ci.yaml

---
 .github/workflows/ci.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 7853d4c2a0..b8d971af63 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -24,7 +24,7 @@ jobs:
           - 'torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2'
           - '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.0 torchvision==0.18.0 torchaudio==2.3.0'
+          - 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
         exclude:
           - py-version: '3.11'
             pytorch-version: 'torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1'