Rename

Quantco · Jun 24, 2024 · dc946dc · dc946dc
1 parent 05787f9
commit dc946dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/metalearners/drlearner.py b/metalearners/drlearner.py
@@ -223,7 +223,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(
@@ -234,7 +234,7 @@ def evaluate(
             model_kind=PROPENSITY_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=False,
+            is_treatment_model=False,
         )
 
         pseudo_outcome: list[np.ndarray] = []
@@ -257,7 +257,7 @@ def evaluate(
             model_kind=TREATMENT_MODEL,
             is_oos=is_oos,
             oos_method=oos_method,
-            is_treatment=True,
+            is_treatment_model=True,
         )
 
         return variant_outcome_evaluation | propensity_evaluation | treatment_evaluation