From 8c14039fdd3ba5348380df947dcfc8d5dd2294f6 Mon Sep 17 00:00:00 2001 From: eroell Date: Sun, 1 Dec 2024 17:09:43 +0100 Subject: [PATCH] correct docstring --- ehrapy/plot/_survival_analysis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ehrapy/plot/_survival_analysis.py b/ehrapy/plot/_survival_analysis.py index b5f88233..717f9202 100644 --- a/ehrapy/plot/_survival_analysis.py +++ b/ehrapy/plot/_survival_analysis.py @@ -231,7 +231,7 @@ def kaplan_meier( >>> adata[:, ["censor_flg"]].X = np.where(adata[:, ["censor_flg"]].X == 0, 1, 0) >>> kmf = ep.tl.kaplan_meier(adata, "mort_day_censored", "censor_flg") - >>> ep.pl.kmf( + >>> ep.pl.kaplan_meier( ... [kmf], color=["r"], xlim=[0, 700], ylim=[0, 1], xlabel="Days", ylabel="Proportion Survived", show=True ... ) @@ -244,7 +244,7 @@ def kaplan_meier( >>> kmf_1 = ep.tl.kaplan_meier(adata_ficu, "mort_day_censored", "censor_flg", label="FICU") >>> kmf_2 = ep.tl.kaplan_meier(adata_micu, "mort_day_censored", "censor_flg", label="MICU") >>> kmf_3 = ep.tl.kaplan_meier(adata_sicu, "mort_day_censored", "censor_flg", label="SICU") - >>> ep.pl.kmf([kmf_1, kmf_2, kmf_3], ci_show=[False,False,False], color=['k','r', 'g'], + >>> ep.pl.kaplan_meier([kmf_1, kmf_2, kmf_3], ci_show=[False,False,False], color=['k','r', 'g'], >>> xlim=[0, 750], ylim=[0, 1], xlabel="Days", ylabel="Proportion Survived") .. image:: /_static/docstring_previews/kmf_plot_2.png