From bd57073985c86e8e51a849261522cfccc414e6c0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?=
 <francesc.marti.escofet@quantco.com>
Date: Wed, 3 Jul 2024 13:50:05 +0200
Subject: [PATCH] Add week units

---
 docs/examples/example_survival.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb
index 8b88fdf..a3fbb6a 100644
--- a/docs/examples/example_survival.ipynb
+++ b/docs/examples/example_survival.ipynb
@@ -182,7 +182,7 @@
     ")\n",
     "plt.hist(simplify_output(cate_estimates_rlearner), bins=20, alpha=0.5, label=\"RLearner\")\n",
     "plt.legend()\n",
-    "plt.xlabel(\"CATE estimate\")\n",
+    "plt.xlabel(\"CATE estimate (weeks)\")\n",
     "plt.ylabel(\"Frequency\")\n",
     "plt.title(\"Histogram of CATE estimates\")\n",
     "plt.show()"