Make the *other* scaler test also more robust (#1123)

wearepal · May 28, 2024 · 344bfeb · 344bfeb
2 parents 6664e95 + 7f3c696
commit 344bfeb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data/dataset_modification_test.py b/tests/data/dataset_modification_test.py
@@ -38,7 +38,7 @@ def test_scaling(
 
     datatuple_post, _ = scale_continuous(dataset, datatuple_scaled, scaler2, inverse=True)
 
-    pd.testing.assert_frame_equal(datatuple.x, datatuple_post.x, check_dtype=False)
+    np.testing.assert_allclose(datatuple.x.to_numpy(), datatuple_post.x.to_numpy(), atol=1e-7)
 
 
 @pytest.mark.parametrize("dataset_name", available_tabular())