From c089c251806e9ef9c5e1e63b801d5195121d19a7 Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Tue, 4 Jun 2024 22:10:05 +0000 Subject: [PATCH] fix: ARIMAPlus loads auto_arima_min_order param --- tests/system/large/ml/test_forecasting.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/system/large/ml/test_forecasting.py b/tests/system/large/ml/test_forecasting.py index 79deb615b1b..438177b1a06 100644 --- a/tests/system/large/ml/test_forecasting.py +++ b/tests/system/large/ml/test_forecasting.py @@ -127,8 +127,7 @@ def test_arima_plus_model_fit_params(time_series_df_default_index, dataset_id): assert reloaded_model.horizon == 100 assert reloaded_model.auto_arima is True assert reloaded_model.auto_arima_max_order == 4 - # TODO(garrettwu): now BQML doesn't populate auto_arima_min_order - # assert reloaded_model.auto_arima_min_order == 1 + assert reloaded_model.auto_arima_min_order == 1 assert reloaded_model.data_frequency == "DAILY" assert reloaded_model.holiday_region == "US" assert reloaded_model.clean_spikes_and_dips is False