@@ -217,7 +217,7 @@ def _setup_test_data_provided(self, stage) -> None:
217217 # Transform the predict data
218218 self .data_predict = self .transform_dataset (self .data_predict )
219219
220- def _setup_data_val_provided (self , stage ) -> None :
220+ def _setup_val_data_provided (self , stage ) -> None :
221221 # New functionality for predefined train, validation and test data in the fun_control
222222 # Get the data set sizes
223223 if self .data_full_train is None :
@@ -229,7 +229,6 @@ def _setup_data_val_provided(self, stage) -> None:
229229 if stage == "fit" or stage is None :
230230 if self .verbosity > 0 :
231231 print (f"train_size: { train_size } , val_size: { val_size } used for train & val data." )
232- generator_fit = torch .Generator ().manual_seed (self .test_seed )
233232 # Use all data from data_full_train as training data
234233 self .data_train = self .data_full_train
235234 # Handle scaling and transformation if scaler is provided
@@ -281,7 +280,7 @@ def setup(self, stage: Optional[str] = None) -> None:
281280 if self .data_full is not None :
282281 self ._setup_full_data_provided (stage )
283282 elif self .data_val is not None :
284- self ._setup_data_val_provided (stage )
283+ self ._setup_val_data_provided (stage )
285284 else :
286285 self ._setup_test_data_provided (stage )
287286
0 commit comments