Skip to content

Commit addd2cf

Browse files
committed
attempt fix test
1 parent 823e3fe commit addd2cf

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

.circleci/config.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -480,7 +480,7 @@ jobs:
480480
- checkout
481481
- run:
482482
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
484484
command: |
485485
python -m venv test_evadb
486486
source test_evadb/bin/activate

evadb/executor/create_function_executor.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,6 @@ def handle_forecasting_function(self):
423423
AutoNHITS,
424424
AutoPatchTST,
425425
AutoTFT,
426-
AutoTimesNet,
427426
)
428427

429428
# from neuralforecast.auto import AutoAutoformer as AutoAFormer
@@ -436,7 +435,6 @@ def handle_forecasting_function(self):
436435
FEDformer,
437436
Informer,
438437
PatchTST,
439-
TimesNet,
440438
)
441439

442440
# from neuralforecast.models import Autoformer as AFormer
@@ -456,8 +454,6 @@ def handle_forecasting_function(self):
456454
# "AutoAFormer": AutoAFormer,
457455
"Informer": Informer,
458456
"AutoInformer": AutoInformer,
459-
"TimesNet": TimesNet,
460-
"AutoTimesNet": AutoTimesNet,
461457
"TFT": TFT,
462458
"AutoTFT": AutoTFT,
463459
}
@@ -546,7 +542,7 @@ def get_optuna_config(trial):
546542
raise FunctionIODefinitionError(err_msg)
547543

548544
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
550546
)
551547

552548
data["ds"] = pd.to_datetime(data["ds"])

test/integration_tests/long/test_model_forecasting.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ def tearDownClass(cls):
7979

8080
@forecast_skip_marker
8181
def test_forecast(self):
82+
8283
create_predict_udf = """
8384
CREATE FUNCTION AirForecast FROM
8485
(SELECT unique_id, ds, y FROM AirData)
@@ -131,7 +132,7 @@ def test_forecast(self):
131132
"airpanelforecast.y-hi",
132133
],
133134
)
134-
135+
135136
@forecast_skip_marker
136137
def test_forecast_with_column_rename(self):
137138
create_predict_udf = """

0 commit comments

Comments
 (0)