Merge branch 'securefederatedai:develop' into straggler_handling_update

securefederatedai · Nov 13, 2024 · 92f8497 · 92f8497
2 parents bd23ceb + b8a1f57
commit 92f8497
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/openfl/federated/data/loader_gandlf.py b/openfl/federated/data/loader_gandlf.py
@@ -25,7 +25,10 @@ def __init__(self, data_path, feature_shape):
             data_path (str): The path to the directory containing the data.
             feature_shape (tuple): The shape of an example feature array.
         """
-        self.train_csv = data_path + "/train.csv"
+        if "inference" in data_path:
+            self.train_csv = None
+        else:
+            self.train_csv = data_path + "/train.csv"
         self.val_csv = data_path + "/valid.csv"
         self.train_dataloader = None
         self.val_dataloader = None