Rename

Quantco · Jun 24, 2024 · c4de4f1 · c4de4f1
1 parent 9d2bbb9
commit c4de4f1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/metalearners/xlearner.py b/metalearners/xlearner.py
@@ -299,7 +299,7 @@ def evaluate(
             model_kind=VARIANT_OUTCOME_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
         propensity_evaluation = _evaluate_model_kind(
@@ -310,7 +310,7 @@ def evaluate(
             model_kind=PROPENSITY_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
         imputed_te_control: list[np.ndarray] = []
@@ -330,7 +330,7 @@ def evaluate(
             model_kind=TREATMENT_EFFECT_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=True,
+            is_treatment_model=True,
         )
 
         te_control_evaluation = _evaluate_model_kind(
@@ -341,7 +341,7 @@ def evaluate(
             model_kind=CONTROL_EFFECT_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=True,
+            is_treatment_model=True,
         )
 
         return (