Rename variables in reuse example. (#95)

Quantco · Dec 3, 2024 · 8b371de · 8b371de
1 parent a4858da
commit 8b371de
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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",