Update test_dl_detector.py

mitre · Dec 27, 2023 · 90871cb · 90871cb
1 parent 8ef1b18
commit 90871cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():