Float weights into survival_table_from_events
Changing survival_table_from_events so it generates float weights and subsequent calculations properly. Edited in reference to Issue
@@ -268,6 +268,9 @@ or "Adjusted Kaplan-Meier estimator and log-rank test with inverse probability o
if collapse:
event_table = _group_event_table_by_intervals(event_table, intervals)
+ if (weights.astype(int) != weights).any():
+ return event_table.astype(float)
+ else:
return event_table.astype(int)
