Rename

Quantco · Jun 24, 2024 · 05787f9 · 05787f9
1 parent d76dc74
commit 05787f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metalearners/metalearner.py b/metalearners/metalearner.py
@@ -158,7 +158,7 @@ def _evaluate_model_kind(
             scorer_name = f"custom_scorer_{idx}"
             scorer_callable = scorer
         for i, cfe in enumerate(cfes):
-            if is_treatment:
+            if is_treatment_model:
                 treatment_variant = i + 1
                 index_str = f"{treatment_variant}_vs_0_"
             else: