From 7fa87940285151fc88b4fd4def4dced461d5b14d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?=
 <154450563+FrancescMartiEscofetQC@users.noreply.github.com>
Date: Mon, 24 Jun 2024 15:40:05 +0200
Subject: [PATCH] Update metalearners/drlearner.py

Co-authored-by: Kevin Klein <7267523+kklein@users.noreply.github.com>
---
 metalearners/drlearner.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/metalearners/drlearner.py b/metalearners/drlearner.py
index 40e8a09..1df7291 100644
--- a/metalearners/drlearner.py
+++ b/metalearners/drlearner.py
@@ -212,9 +212,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)],