Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit 43d4ae0

Browse files
committed
test(ml): Update arima plus score tests to ignore extra BQML columns
1 parent edceb35 commit 43d4ae0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/system/small/ml/test_forecasting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def test_arima_plus_score(
493493
dtype="Float64",
494494
)
495495
pd.testing.assert_frame_equal(
496-
result,
496+
result[expected.columns],
497497
expected,
498498
rtol=0.1,
499499
check_index_type=False,
@@ -594,7 +594,7 @@ def test_arima_plus_score_series(
594594
dtype="Float64",
595595
)
596596
pd.testing.assert_frame_equal(
597-
result,
597+
result[expected.columns],
598598
expected,
599599
rtol=0.1,
600600
check_index_type=False,

0 commit comments

Comments
 (0)