From e81d152172c3a1d92824f83ea4467f3defc937ca 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:35:58 +0200
Subject: [PATCH] Rename

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

diff --git a/metalearners/slearner.py b/metalearners/slearner.py
index aacc93b..4e3e2b7 100644
--- a/metalearners/slearner.py
+++ b/metalearners/slearner.py
@@ -172,7 +172,7 @@ def evaluate(
             model_kind=_BASE_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
     def predict_conditional_average_outcomes(