Update __init__.py
Corrected if-then logic
@@ -268,7 +268,7 @@ 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():
+ if (np.asarray(weights).astype(int) != weights).any():
return event_table.astype(float)
else:
return event_table.astype(int)
