fix type error

SeldonIO · Oct 24, 2022 · 1fa1610 · 1fa1610
1 parent 2cce862
commit 1fa1610
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/alibi_detect/cd/sklearn/classifier.py b/alibi_detect/cd/sklearn/classifier.py
@@ -229,7 +229,8 @@ def predict_proba(self, X):
 
         return model
 
-    def score(self, x: Union[np.ndarray, list]) -> Tuple[float, float, np.ndarray, np.ndarray, np.ndarray, np.ndarray]:
+    def score(self, x: Union[np.ndarray, list]) \
+            -> Tuple[float, float, np.ndarray, np.ndarray, Union[np.ndarray, list], Union[np.ndarray, list]]:
         """
         Compute the out-of-fold drift metric such as the accuracy from a classifier
         trained to distinguish the reference data from the data to be tested.