Updated function name

burggraaff · Jun 4, 2024 · d3ac2a5 · d3ac2a5
1 parent b02b441
commit d3ac2a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fpcup/nn/dataset.py b/fpcup/nn/dataset.py
@@ -93,8 +93,8 @@ def __init__(self, data_dir: PathOrStr, *, transform=None, target_transform=None
 
         # Load data
         self.summary = load_combined_ensemble_summary(data_dir, pattern=pattern, save_if_generated=False, **kwargs)
-        add_input_columns(self.summary)
-        add_output_columns(self.summary)
+        preprocess_X(self.summary)
+        preprocess_y(self.summary)
         self.summary = self.summary[INPUTS + OUTPUTS]
 
     def __len__(self) -> int: