From ce8c63f52a120576aad61fe0c7274033c3bc5639 Mon Sep 17 00:00:00 2001
From: Martin Bernstorff <ryqiem@gmail.com>
Date: Tue, 9 May 2023 10:18:46 +0200
Subject: [PATCH] feat: add ci to timedelta plots

---
 src/psycop/model_evaluation/binary/time/timedelta_plots.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/psycop/model_evaluation/binary/time/timedelta_plots.py b/src/psycop/model_evaluation/binary/time/timedelta_plots.py
index 35e3221f9..3bd7e58fc 100644
--- a/src/psycop/model_evaluation/binary/time/timedelta_plots.py
+++ b/src/psycop/model_evaluation/binary/time/timedelta_plots.py
@@ -69,8 +69,11 @@ def plot_roc_auc_by_time_from_first_visit(
         bin_unit=bin_unit,
         bin_continuous_input=bin_continuous_input,
         drop_na_events=False,
+        confidence_interval=0.95,
     )
 
+    ci = df["ci"].tolist()
+
     bin_unit2str = {
         "h": "Hours",
         "D": "Days",
@@ -90,6 +93,7 @@ def plot_roc_auc_by_time_from_first_visit(
         plot_type=["line", "scatter"],
         bar_count_values=df["n_in_bin"],
         bar_count_y_axis_title="Number of visits",
+        confidence_interval=ci,
         save_path=save_path,
     )