Add warning to plot_feature_outier_image

SeldonIO · Apr 24, 2023 · 6408792 · 6408792
1 parent 9d10d24
commit 6408792
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/alibi_detect/utils/visualize.py b/alibi_detect/utils/visualize.py
@@ -3,6 +3,7 @@
 import pandas as pd
 from sklearn.metrics import roc_curve, auc
 from typing import Dict, Union
+import warnings
 
 
 def plot_instance_score(preds: Dict,
@@ -79,6 +80,7 @@ def plot_feature_outlier_image(od_preds: Dict,
     instance_ids = instance_ids[:n_instances]
 
     if outliers_only and n_instances == 0:
+        warnings.warn('No outliers found!', UserWarning, stacklevel=3)
         return
 
     n_cols = 2