Update metalearners/metalearner.py

Co-authored-by: Francesc Martí Escofet <[email protected]>
Quantco · Jun 24, 2024 · 03f032b · 03f032b
1 parent ee8c944
commit 03f032b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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(