Update test_psi_detector.py

mitre · Nov 22, 2023 · 46c72e7 · 46c72e7
1 parent 3db9642
commit 46c72e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/menelaus/data_drift/test_psi_detector.py b/tests/menelaus/data_drift/test_psi_detector.py
@@ -1,6 +1,6 @@
 import pytest
 import numpy as np
-from menelaus.data_drift.psi_detector import PSI
+from menelaus.data_drift import PSI
 
 def test_psi_init():
     """Test correct default initialization for PSI"""