Skip to content

Commit 2997f55

Browse files
committed
update lightdatamodule.py
1 parent d0ba56f commit 2997f55

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/spotpython/data/lightdatamodule.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ def __init__(
6363
dataset: Optional[object] = None,
6464
data_full_train: Optional[object] = None,
6565
data_test: Optional[object] = None,
66+
data_val: Optional[object] = None,
6667
test_size: Optional[float] = None,
6768
test_seed: int = 42,
6869
num_workers: int = 0,
@@ -74,6 +75,7 @@ def __init__(
7475
self.data_full = dataset
7576
self.data_full_train = data_full_train
7677
self.data_test = data_test
78+
self.data_val = data_val
7779
self.test_size = test_size
7880
self.test_seed = test_seed
7981
self.num_workers = num_workers

0 commit comments

Comments
 (0)