File tree Expand file tree Collapse file tree 3 files changed +4
-7
lines changed
test/integration_tests/long Expand file tree Collapse file tree 3 files changed +4
-7
lines changed Original file line number Diff line number Diff line change @@ -480,7 +480,7 @@ jobs:
480
480
- checkout
481
481
- run :
482
482
name : Install EvaDB package from GitHub repo and run tests
483
- no_output_timeout : 40m # 40 minute timeout
483
+ no_output_timeout : 30m # 30 minute timeout
484
484
command : |
485
485
python -m venv test_evadb
486
486
source test_evadb/bin/activate
Original file line number Diff line number Diff line change @@ -423,7 +423,6 @@ def handle_forecasting_function(self):
423
423
AutoNHITS ,
424
424
AutoPatchTST ,
425
425
AutoTFT ,
426
- AutoTimesNet ,
427
426
)
428
427
429
428
# from neuralforecast.auto import AutoAutoformer as AutoAFormer
@@ -436,7 +435,6 @@ def handle_forecasting_function(self):
436
435
FEDformer ,
437
436
Informer ,
438
437
PatchTST ,
439
- TimesNet ,
440
438
)
441
439
442
440
# from neuralforecast.models import Autoformer as AFormer
@@ -456,8 +454,6 @@ def handle_forecasting_function(self):
456
454
# "AutoAFormer": AutoAFormer,
457
455
"Informer" : Informer ,
458
456
"AutoInformer" : AutoInformer ,
459
- "TimesNet" : TimesNet ,
460
- "AutoTimesNet" : AutoTimesNet ,
461
457
"TFT" : TFT ,
462
458
"AutoTFT" : AutoTFT ,
463
459
}
@@ -546,7 +542,7 @@ def get_optuna_config(trial):
546
542
raise FunctionIODefinitionError (err_msg )
547
543
548
544
model = StatsForecast (
549
- [model_here (season_length = season_length )], freq = new_freq , n_jobs = - 1
545
+ [model_here (season_length = season_length )], freq = new_freq
550
546
)
551
547
552
548
data ["ds" ] = pd .to_datetime (data ["ds" ])
Original file line number Diff line number Diff line change @@ -79,6 +79,7 @@ def tearDownClass(cls):
79
79
80
80
@forecast_skip_marker
81
81
def test_forecast (self ):
82
+
82
83
create_predict_udf = """
83
84
CREATE FUNCTION AirForecast FROM
84
85
(SELECT unique_id, ds, y FROM AirData)
@@ -131,7 +132,7 @@ def test_forecast(self):
131
132
"airpanelforecast.y-hi" ,
132
133
],
133
134
)
134
-
135
+
135
136
@forecast_skip_marker
136
137
def test_forecast_with_column_rename (self ):
137
138
create_predict_udf = """
You can’t perform that action at this time.
0 commit comments