From 76e67a9e794333994132008bf87beeb54337be7c Mon Sep 17 00:00:00 2001
From: Rantaharju Jarno <jarno.rantaharju@aalto.fi>
Date: Mon, 9 Dec 2024 13:26:18 +0200
Subject: [PATCH] Select columns in local_time

---
 niimpy/preprocessing/location.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/niimpy/preprocessing/location.py b/niimpy/preprocessing/location.py
index d94973df..ff8b7379 100644
--- 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