Select columns in local_time

digitraceslab · Dec 9, 2024 · 76e67a9 · 76e67a9
1 parent 6f80707
commit 76e67a9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/niimpy/preprocessing/location.py b/niimpy/preprocessing/location.py
@@ -520,6 +520,7 @@ def set_timezone(row):
     df = util.reset_groups(df)
     df["timezone"] = df.apply(get_timezone, axis=1)
     df = df.apply(set_timezone, axis=1)
+    df = util.select_columns(df, ["timezone"])
     return df