Rename

Quantco · Jun 24, 2024 · ba895a3 · ba895a3
1 parent dc946dc
commit ba895a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metalearners/rlearner.py b/metalearners/rlearner.py
@@ -349,7 +349,7 @@ def evaluate(
             model_kind=PROPENSITY_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
         outcome_evaluation = _evaluate_model_kind(
@@ -360,7 +360,7 @@ def evaluate(
             model_kind=OUTCOME_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
         # TODO: improve this? generalize it to other metalearners?