Update metalearners/xlearner.py

Co-authored-by: Kevin Klein <[email protected]>
Quantco · Jun 24, 2024 · 5f0987f · 5f0987f
1 parent 4a36e25
commit 5f0987f
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/metalearners/xlearner.py b/metalearners/xlearner.py
@@ -291,10 +291,6 @@ def evaluate(
     ) -> dict[str, float]:
         safe_scoring = self._scoring(scoring)
 
-        masks = []
-        for tv in range(self.n_variants):
-            masks.append(w == tv)
-
         variant_outcome_evaluation = _evaluate_model_kind(
             cfes=self._nuisance_models[VARIANT_OUTCOME_MODEL],
             Xs=[X[w == tv] for tv in range(self.n_variants)],