fix imports

scaleoutsystems · May 7, 2024 · 8ebe64d · 8ebe64d
1 parent 7a5197e
commit 8ebe64d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/mnist-pytorch/client/train.py b/examples/mnist-pytorch/client/train.py
@@ -5,9 +5,10 @@
 import torch
 from model import load_parameters, save_parameters
 
-from data import load_data
 from fedn.utils.helpers.helpers import save_metadata
 
+from .data import load_data
+
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(os.path.abspath(dir_path))
 

diff --git a/examples/mnist-pytorch/client/validate.py b/examples/mnist-pytorch/client/validate.py
@@ -4,9 +4,10 @@
 import torch
 from model import load_parameters
 
-from data import load_data
 from fedn.utils.helpers.helpers import save_metrics
 
+from .data import load_data
+
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.append(os.path.abspath(dir_path))