correct test

DataResponsibly · Dec 3, 2024 · 077c679 · 077c679
1 parent 4d47e2e
commit 077c679
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sharp/visualization/tests/test_visualization.py b/sharp/visualization/tests/test_visualization.py
@@ -13,8 +13,9 @@ def mock_sharp():
 
     class MockSharp:
         def __init__(self):
-            self.qoi = "rank"
-            self.measure = "shapley"
+            self.feature_names_ = np.array(["Feature1", "Feature2", "Feature3"])
+            self.qoi="rank"
+            self.measure="shapley"
 
     return MockSharp()