Skip to content

Commit 1acd75c

Browse files
committed
Model test
1 parent 1537871 commit 1acd75c

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

tests/test_model.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -100,11 +100,13 @@ def test_bris_predictor():
100100
"%Y-%m-%dT%H:%M:%S",
101101
)
102102

103-
config.dataset = {
104-
"dataset": config.dataset,
105-
"start": config.start_date,
106-
"end": config.end_date,
107-
"frequency": config.frequency,
103+
config.datasets = {
104+
"data": {
105+
"dataset": config.datasets.data,
106+
"start": config.start_date,
107+
"end": config.end_date,
108+
"frequency": config.frequency,
109+
}
108110
}
109111

110112
datamodule = DataModule(

0 commit comments

Comments
 (0)