Determine correct stattest for descriptors. (#1164)

evidentlyai · Jun 20, 2024 · 3fee32d · 3fee32d
1 parent b30747c
commit 3fee32d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/evidently/metrics/data_drift/column_drift_metric.py b/src/evidently/metrics/data_drift/column_drift_metric.py
@@ -56,18 +56,17 @@ def get_one_column_drift(
 
     target = data_definition.get_target_column()
     stattest = None
-    threshold = None
     if column.is_main_dataset():
         if target and column.name == target.column_name and column_type == ColumnType.Numerical:
             stattest = options.num_target_stattest_func
 
         elif target and column.name == target.column_name and column_type == ColumnType.Categorical:
             stattest = options.cat_target_stattest_func
 
-        if not stattest:
-            stattest = options.get_feature_stattest_func(column.name, column_type.value)
+    if not stattest:
+        stattest = options.get_feature_stattest_func(column.name, column_type.value)
 
-        threshold = options.get_threshold(column.name, column_type.value)
+    threshold = options.get_threshold(column.name, column_type.value)
     current_column = current_feature_data
     reference_column = reference_feature_data