small fix of test_patch_tst

aimclub · Oct 16, 2024 · 342860d · 342860d
1 parent 257b24a
commit 342860d
Showing 1 changed file with 4 additions and 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
@@ -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)