added stacklevel to deprecation warning

theislab · Nov 27, 2024 · c604074 · c604074
1 parent 82a6e3c
commit c604074
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ehrapy/tools/_sa.py b/ehrapy/tools/_sa.py
@@ -163,6 +163,7 @@ def kmf(
     warnings.warn(
         "This function is deprecated and will be removed in the next release. Use `kaplan_meier` instead.",
         DeprecationWarning,
+        stacklevel=2
     )
     kmf = KaplanMeierFitter()
     if censoring == "None" or "right":