Update test_psi_detector.py

mitre · Nov 22, 2023 · 5d9f06f · 5d9f06f
1 parent cc85fac
commit 5d9f06f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/menelaus/data_drift/test_psi_detector.py b/tests/menelaus/data_drift/test_psi_detector.py
@@ -57,5 +57,4 @@ def test_psi_compute_PSI():
     test = np.random.randint(0,100,100)
     det.set_reference(X = ref)
     det.update(X = test)
-    threshold = det.PSI_value
-    assert threshold >= 0 and threshold <= 1
+    assert det.PSI_value >= 0 and det.PSI_value <= 1