monkey patch added

Signed-off-by: yes <[email protected]>
securefederatedai · Dec 23, 2024 · 5a2b163 · 5a2b163
1 parent 8896b28
commit 5a2b163
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/openfl/federated/data/loader_pt.py b/openfl/federated/data/loader_pt.py
@@ -4,12 +4,16 @@
 
 """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.