Fix test

MeteoSwiss · Sep 6, 2024 · 8fd19ad · 8fd19ad
1 parent 9f06ade
commit 8fd19ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_train.py b/tests/test_train.py
@@ -158,7 +158,7 @@ def test_train_fromfile(tmp_path, cfg):
     datasplitter = DataSplitter(
         splitter_options.time_split, splitter_options.station_split
     )
-    datanormalizer = DataTransformer(**["normalizer"])
+    datanormalizer = DataTransformer(**cfg["normalizer"])
     batch_dims = ["forecast_reference_time", "t", "station"]
     datamodule = DataModule(
         features=cfg["features"],