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