[FeTS][GaNDLF] skip loading train.csv during inference mode (secure…

…federatedai#1141) * remove train.csv path during inference step Signed-off-by: kta-intel <[email protected]> * check inference during dataloader Signed-off-by: kta-intel <[email protected]> * lint fix Signed-off-by: kta-intel <[email protected]> --------- Signed-off-by: kta-intel <[email protected]>
tanwarsh · Nov 18, 2024 · b4a384b · b4a384b
1 parent 31ce02d
commit b4a384b
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