owkaplanmeier: cast to a float before fitting the Kaplan-Meier curve.

biolab · Jan 12, 2021 · 6d20119 · 6d20119
1 parent 1fe2cd3
commit 6d20119
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orangecontrib/survival_analysis/widgets/owkaplanmeier.py b/orangecontrib/survival_analysis/widgets/owkaplanmeier.py
@@ -35,7 +35,7 @@ def generate_curve_coordinates(timeline, probabilities):
         return np.array(x), np.array(y)
 
     def __init__(self, time, events, label=None, color=None):
-        self._kmf = KaplanMeierFitter().fit(time, events)
+        self._kmf = KaplanMeierFitter().fit(time.astype(np.float64), events.astype(np.float64))
 
         self.label: str = label
         self.color: List[int] = color