From 6f35718de2cf374c2a13beb601b0871e2c9ba2fc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= Date: Wed, 3 Jul 2024 14:06:24 +0200 Subject: [PATCH] Fix mypy errors --- docs/examples/example_survival.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb index e4a2158..486349f 100644 --- 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",