Update metalearners/metalearner.py

Co-authored-by: Francesc Martí Escofet <[email protected]>
Quantco · Jul 25, 2024 · 4cc4f68 · 4cc4f68
1 parent 474c598
commit 4cc4f68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metalearners/metalearner.py b/metalearners/metalearner.py
@@ -1163,7 +1163,7 @@ def init_args(self) -> dict[str, Any]:
                 if k in self._prefitted_nuisance_models and k != PROPENSITY_MODEL
             },
             "fitted_propensity_model": (
-                self._nuisance_models.get(PROPENSITY_MODEL)
+                deepcopy(self._nuisance_models.get(PROPENSITY_MODEL))
                 if PROPENSITY_MODEL in self._prefitted_nuisance_models
                 else None
             ),