From 05787f9d1652d34b9bd32bd9d16646bb547f143b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?=
 <154450563+FrancescMartiEscofetQC@users.noreply.github.com>
Date: Mon, 24 Jun 2024 13:32:32 +0200
Subject: [PATCH] Rename

---
 metalearners/metalearner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metalearners/metalearner.py b/metalearners/metalearner.py
index 6c1604a..42c4377 100644
--- 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: