diff --git a/docs/examples/example_reuse.ipynb b/docs/examples/example_reuse.ipynb index 4d49768..4a9b655 100644 --- a/docs/examples/example_reuse.ipynb +++ b/docs/examples/example_reuse.ipynb @@ -175,7 +175,15 @@ }, "outputs": [], "source": [ - "outcome_model = rlearner._nuisance_models[\"outcome_model\"]" + "outcome_models = rlearner._nuisance_models[\"outcome_model\"]\n", + "outcome_models" + ] + }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "Note that ``outcome_models`` is a sequence of models - in this case of length 1." ] }, { @@ -211,7 +219,7 @@ " propensity_model_factory=LogisticRegression,\n", " treatment_model_factory=LinearRegression,\n", " is_classification=False,\n", - " fitted_nuisance_models={\"outcome_model\": outcome_model},\n", + " fitted_nuisance_models={\"outcome_model\": outcome_models},\n", " propensity_model_params={\"max_iter\": 500},\n", " n_variants=2,\n", ")\n",