Improved test coverage

unit8co · Oct 28, 2024 · ebe02d1 · ebe02d1
1 parent 3a392a3
commit ebe02d1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/darts/tests/models/forecasting/test_tsmixer.py b/darts/tests/models/forecasting/test_tsmixer.py
@@ -365,7 +365,7 @@ def test_time_batch_norm_2d_gradients(self):
 
     @pytest.mark.parametrize("project_first_layer", [True, False])
     def test_project_first(self, project_first_layer):
-        ts = tg.sine_timeseries(length=36)
+        ts = tg.sine_timeseries(length=36, freq="h")
         input_len = 12
         output_len = 6
 
@@ -374,6 +374,8 @@ def test_project_first(self, project_first_layer):
             output_chunk_length=output_len,
             n_epochs=1,
             project_first_layer=project_first_layer,
+            # Cover case of projecting future covs back to input dims
+            add_encoders={"cyclic": {"future": "hour"}},
             **tfm_kwargs,
         )
         model.fit(ts)