Fix feature name

nubank · Aug 15, 2023 · ff7ae18 · ff7ae18
1 parent b154784
commit ff7ae18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/metrics/test_pd_extractors.py b/tests/metrics/test_pd_extractors.py
@@ -158,7 +158,7 @@ def test_extract():
         spearman_evaluator(target_column='target', prediction_column='prediction')
     ])
 
-    splitter = split_evaluator(eval_fn=base_evaluator, split_col='median_income', split_values=[0.5, 10.0, 20.0])
+    splitter = split_evaluator(eval_fn=base_evaluator, split_col='MedInc', split_values=[0.5, 10.0, 20.0])
     temporal_week_splitter = temporal_split_evaluator(eval_fn=base_evaluator, time_col='time', time_format='%Y-%W')
     temporal_year_splitter = temporal_split_evaluator(eval_fn=base_evaluator, time_col='time', time_format='%Y')