Update metalearners/xlearner.py

Co-authored-by: Matthias Loeffler <[email protected]>
Quantco · Aug 16, 2024 · 6803096 · 6803096
1 parent c8dd77e
commit 6803096
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metalearners/xlearner.py b/metalearners/xlearner.py
@@ -557,7 +557,7 @@ def predict_conditional_average_outcomes(
                         X=X,
                         model_kind=VARIANT_OUTCOME_MODEL,
                         model_ord=tv,
-                        is_oos=True,
+                        is_oos=is_oos,
                         oos_method=oos_method,
                     )
                 )