diff --git a/metalearners/metalearner.py b/metalearners/metalearner.py index d9a7d63..13d9311 100644 --- a/metalearners/metalearner.py +++ b/metalearners/metalearner.py @@ -1158,7 +1158,7 @@ def init_args(self) -> dict[str, Any]: else None ), "fitted_nuisance_models": { - k: v + k: deepcopy(v) for k, v in self._nuisance_models.items() if k in self._prefitted_nuisance_models and k != PROPENSITY_MODEL },