From 90871cbcb49a552a82329bc4810fafca13d764c3 Mon Sep 17 00:00:00 2001 From: Tim Chen <115333718+951378644@users.noreply.github.com> Date: Wed, 27 Dec 2023 16:27:08 -0500 Subject: [PATCH] Update test_dl_detector.py --- tests/menelaus/data_drift/test_dl_detector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/menelaus/data_drift/test_dl_detector.py b/tests/menelaus/data_drift/test_dl_detector.py index efc62ba..effc3eb 100644 --- a/tests/menelaus/data_drift/test_dl_detector.py +++ b/tests/menelaus/data_drift/test_dl_detector.py @@ -25,7 +25,7 @@ def test_dl_detector_set_reference(): ) det.set_reference(ref) scale_ref = pd.DataFrame(scaler.fit_transform(ref), columns=ref.columns, index=ref.index) - assert det.reference == scale_ref + pd.testing.assert_frame_equal(det.reference, scale_ref) def test_dl_detector_update_1():