fix more tests

tswast · tswast · commit 5ab21a9cc199 · 2025-08-21T23:12:59.000Z
diff --git a/tests/system/small/ml/test_forecasting.py b/tests/system/small/ml/test_forecasting.py
@@ -490,6 +490,7 @@ def test_arima_plus_score(
         expected.sort_values("id").reset_index(),
         rtol=0.1,
         check_index_type=False,
+        check_dtype=False,
     )
 
 
@@ -583,6 +584,7 @@ def test_arima_plus_score_series(
         expected.sort_values("id").reset_index(),
         rtol=0.1,
         check_index_type=False,
+        check_dtype=False,
     )
 
 
diff --git a/tests/system/small/ml/test_preprocessing.py b/tests/system/small/ml/test_preprocessing.py
@@ -245,7 +245,7 @@ def test_max_abs_scaler_save_load(new_penguins_df, dataset_id):
         index=pd.Index([1633, 1672, 1690], name="tag_number", dtype="Int64"),
     )
 
-    pd.testing.assert_frame_equal(result, expected, rtol=0.1)
+    pd.testing.assert_frame_equal(result.sort_index(), expected.sort_index(), rtol=0.1)
 
 
 def test_min_max_scaler_normalized_fit_transform(new_penguins_df):

Original file line number	Diff line number	Diff line change
`@@ -490,6 +490,7 @@ def test_arima_plus_score(`
`490`	`490`	`expected.sort_values("id").reset_index(),`
`491`	`491`	`rtol=0.1,`
`492`	`492`	`check_index_type=False,`
	`493`	`+ check_dtype=False,`
`493`	`494`	`)`
`494`	`495`
`495`	`496`
`@@ -583,6 +584,7 @@ def test_arima_plus_score_series(`
`583`	`584`	`expected.sort_values("id").reset_index(),`
`584`	`585`	`rtol=0.1,`
`585`	`586`	`check_index_type=False,`
	`587`	`+ check_dtype=False,`
`586`	`588`	`)`
`587`	`589`
`588`	`590`
Original file line number	Diff line number	Diff line change
`@@ -245,7 +245,7 @@ def test_max_abs_scaler_save_load(new_penguins_df, dataset_id):`
`245`	`245`	`index=pd.Index([1633, 1672, 1690], name="tag_number", dtype="Int64"),`
`246`	`246`	`)`
`247`	`247`
`248`		`- pd.testing.assert_frame_equal(result, expected, rtol=0.1)`
	`248`	`+ pd.testing.assert_frame_equal(result.sort_index(), expected.sort_index(), rtol=0.1)`
`249`	`249`
`250`	`250`
`251`	`251`	`def test_min_max_scaler_normalized_fit_transform(new_penguins_df):`