Fix mypy errors

Quantco · Jul 3, 2024 · 6f35718 · 6f35718
1 parent a91ac2a
commit 6f35718
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb
@@ -185,9 +185,9 @@
     "    bins=40,\n",
     ")[1]\n",
     "\n",
-    "plt.hist(cate_estimates_tlearner, bins=bins, alpha=0.5, label=\"TLearner\")\n",
-    "plt.hist(cate_estimates_drlearner, bins=bins, alpha=0.5, label=\"DRLearner\")\n",
-    "plt.hist(cate_estimates_rlearner, bins=bins, alpha=0.5, label=\"RLearner\")\n",
+    "plt.hist(cate_estimates_tlearner, bins=list(bins), alpha=0.5, label=\"TLearner\")\n",
+    "plt.hist(cate_estimates_drlearner, bins=list(bins), alpha=0.5, label=\"DRLearner\")\n",
+    "plt.hist(cate_estimates_rlearner, bins=list(bins), alpha=0.5, label=\"RLearner\")\n",
     "plt.legend()\n",
     "plt.xlabel(\"CATE estimate (weeks)\")\n",
     "plt.ylabel(\"Frequency\")\n",