From 675d3a18af5fe6f8e804db1891e51a06a77f4765 Mon Sep 17 00:00:00 2001
From: ZhixiaoSu <37242111+ZhixiaoSu@users.noreply.github.com>
Date: Thu, 31 Oct 2024 04:13:18 -0700
Subject: [PATCH] Change absolute time reference to first go cue

---
 src/aind_dynamic_foraging_data_utils/nwb_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aind_dynamic_foraging_data_utils/nwb_utils.py b/src/aind_dynamic_foraging_data_utils/nwb_utils.py
index fec54c6..b23b6e3 100644
--- a/src/aind_dynamic_foraging_data_utils/nwb_utils.py
+++ b/src/aind_dynamic_foraging_data_utils/nwb_utils.py
@@ -329,7 +329,7 @@ def create_df_trials(nwb_filename):
     df_ses_trials["ses_idx"] = ses_idx
 
     # Adjust all times relative to start of the first trial
-    t0 = df_ses_trials.start_time[0]
+    t0 = df_ses_trials.goCue_start_time[0]
     skip_cols = ["right_valve_open_time", "left_valve_open_time"]
     for col in df_ses_trials.columns:
         if ("time" in col) and (col not in skip_cols):