Skip to content

Commit 86475d3

Browse files
committed
added tests for theta forecaster
1 parent 687902d commit 86475d3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/operators/forecast/test_datasets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ def test_pandas_to_historical(model):
242242
check_output_for_errors(output_data_path)
243243

244244

245-
@pytest.mark.parametrize("model", ["prophet", "neuralprophet", "theta"])
245+
@pytest.mark.parametrize("model", ["prophet", "neuralprophet"])
246246
def test_pandas_to_historical_test(model):
247247
df = pd.read_csv(f"{DATASET_PREFIX}dataset4.csv")
248248
df_train = df[:-PERIODS]
@@ -269,7 +269,7 @@ def test_pandas_to_historical_test(model):
269269

270270

271271
# CostAD
272-
@pytest.mark.parametrize("model", ["prophet", "neuralprophet", "theta"])
272+
@pytest.mark.parametrize("model", ["prophet", "neuralprophet"])
273273
def test_pandas_to_historical_test2(model):
274274
df = pd.read_csv(f"{DATASET_PREFIX}dataset5.csv")
275275
df_train = df[:-1]

0 commit comments

Comments
 (0)