fix ttb bug (#411)

gelles-brandeis · Dec 16, 2022 · 05bb6e8 · 05bb6e8
1 parent ea86d8c
commit 05bb6e8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tapqir/imscroll/glimpse_reader.py b/tapqir/imscroll/glimpse_reader.py
@@ -101,11 +101,11 @@ def __init__(self, c=0, **kwargs):
 
         # calculate the cumulative sum of dx and dy relative to the aoiinfo frame
         aoiinfo_frame = int(aoi_df["ontarget"].at[1, "frame"])
-        drift_df.loc[aoiinfo_frame + 1 :] = (
-            drift_df.loc[aoiinfo_frame + 1 :].cumsum(axis=0).values
+        drift_df.loc[aoiinfo_frame + 1 :, ["dx", "dy"]] = (
+            drift_df.loc[aoiinfo_frame + 1 :, ["dx", "dy"]].cumsum(axis=0).values
         )
-        drift_df.loc[aoiinfo_frame - 1 :: -1] = (
-            (-drift_df.loc[aoiinfo_frame : drift_df.index[1] : -1])
+        drift_df.loc[aoiinfo_frame - 1 :: -1, ["dx", "dy"]] = (
+            (-drift_df.loc[aoiinfo_frame : drift_df.index[1] : -1, ["dx", "dy"]])
             .cumsum(axis=0)
             .values
         )