From 342860dfb838a79dda4b23a3e83d2c9874be96ae Mon Sep 17 00:00:00 2001 From: technocreep Date: Wed, 16 Oct 2024 16:43:05 +0200 Subject: [PATCH] small fix of test_patch_tst --- tests/unit/core/models/model_impl/test_patch_tst.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/core/models/model_impl/test_patch_tst.py b/tests/unit/core/models/model_impl/test_patch_tst.py index efca456c1..075783d2d 100644 --- a/tests/unit/core/models/model_impl/test_patch_tst.py +++ b/tests/unit/core/models/model_impl/test_patch_tst.py @@ -23,13 +23,13 @@ def ts_input_data(): return train_test_data_setup(train_input, validation_blocks=None) -def test_nbeats_model(ts_input_data): +def test_patch_tst_model(ts_input_data): train, test = ts_input_data with IndustrialModels(): - model = PipelineBuilder().add_node('patch_tst_model', params=dict( - epochs=10 - )).build() + model = PipelineBuilder().add_node('patch_tst_model', + params=dict(epochs=10) + ).build() model.fit(train) forecast = model.predict(test)