From 00ed38d47df2cc0eae9d1c36e31ac6b750257f76 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Sun, 31 Dec 2023 18:33:44 +1100
Subject: [PATCH] Fix error in survtime_to_numeric when applied to 1D data

---
 yli/survival.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yli/survival.py b/yli/survival.py
index 23457e5..e3456ff 100644
--- a/yli/survival.py
+++ b/yli/survival.py
@@ -331,7 +331,7 @@ def survtime_to_numeric(df, time, time2=None):
 		df[time] = df[time].dt.total_seconds()
 		max_time = df[time].max()
 	
-	if df[time2].dtype == '<m8[ns]':
+	if time2 and df[time2].dtype == '<m8[ns]':
 		df[time2] = df[time2].dt.total_seconds()
 		max_time = max(max_time or 0, df[time2].max())