changes for torch version 2.4.1

Signed-off-by: yes <[email protected]>

openfl-workspace/torch_cnn_histology/requirements.txt

@@ -1,3 +1,2 @@
-setuptools==59.6.0
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_histology_fedcurv/requirements.txt

@@ -1,2 +1,2 @@
-torch==2.3.1 
-torchvision==0.18.1 
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_mnist/requirements.txt

@@ -1,3 +1,2 @@
-setuptools==59.6.0
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_mnist_eden_compression/requirements.txt

@@ -1,5 +1,4 @@
-setuptools==59.6.0
 tensorboard
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_cnn_mnist_fed_eval/requirements.txt

@@ -1,5 +1,4 @@
-setuptools==59.6.0
 tensorboard
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_cnn_mnist_straggler_check/requirements.txt

@@ -1,7 +1,7 @@
 numpy>=1.22.2 # not directly required, pinned by Snyk to avoid a vulnerability
 rsa>=4.7 # not directly required, pinned by Snyk to avoid a vulnerability
-setuptools==59.6.0
+setuptools>=65.5.1 # not directly required, pinned by Snyk to avoid a vulnerability
 tensorboard
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_llm_horovod/requirements.txt

@@ -7,9 +7,8 @@ jupyter
 peft
 sentencepiece
 seqeval
-setuptools==59.6.0
 tensorboard
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1
 transformers[torch]
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_template/requirements.txt

@@ -1,4 +1,3 @@
 # Set your workspace's requirements here
-setuptools==59.6.0
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_unet_kvasir/requirements.txt

@@ -1,4 +1,3 @@
 scikit-image==0.24.0
-setuptools==59.6.0
-torch==2.5.1
-torchvision==0.20.1
+torch==2.4.1
+torchvision==0.19.1

openfl/federated/data/loader_pt.py

@@ -4,16 +4,12 @@
 
 """PyTorchDataLoader module."""
 
-import sys
 from math import ceil
 
 import numpy as np
-import typing_extensions
 
 from openfl.federated.data.loader import DataLoader
 
-sys.modules["pip._vendor.typing_extensions"] = typing_extensions
-
 
 class PyTorchDataLoader(DataLoader):
     """A class used to represent a Federation Data Loader for PyTorch models.