From 03f032b7fdf7b5c516f504df833bba71bb37eed2 Mon Sep 17 00:00:00 2001
From: Kevin Klein <7267523+kklein@users.noreply.github.com>
Date: Mon, 24 Jun 2024 12:12:48 +0200
Subject: [PATCH] Update metalearners/metalearner.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Francesc Martí Escofet <154450563+FrancescMartiEscofetQC@users.noreply.github.com>
---
 metalearners/metalearner.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/metalearners/metalearner.py b/metalearners/metalearner.py
index bb5546e..8aa86a6 100644
--- a/metalearners/metalearner.py
+++ b/metalearners/metalearner.py
@@ -1033,12 +1033,10 @@ def _default_scoring() -> Scoring:
             return {
                 nuisance_model: [
                     default_metric(
-                        self.nuisance_model_specifications()[nuisance_model][
-                            "predict_method"
-                        ](self)
+                        model_specifications["predict_method"](self)
                     )
                 ]
-                for nuisance_model in self.nuisance_model_specifications()
+                for nuisance_model, model_specifications in self.nuisance_model_specifications().items()
             } | {
                 treatment_model: [
                     default_metric(