Ensure longitude is in data columns before type checking

onnela-lab · Jan 31, 2024 · a1eb117 · a1eb117
1 parent ccf30f7
commit a1eb117
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/forest/jasmine/traj2stats.py b/forest/jasmine/traj2stats.py
@@ -1661,8 +1661,13 @@ def gps_stats_main(
                 tz_str, time_start, time_end,
             )
 
-            if ((data["longitude"].max() > 180)
-                    or (data["longitude"].min() < -180)):
+            if (
+                    ("longitude" in data.columns)
+                    and (
+                        (data["longitude"].max() > 180)
+                        or (data["longitude"].min() < -180)
+                        )
+                    ):
                 logger.info("Reconciled bad longitude data for user %s",
                             participant_id)
                 data["longitude"] = (data["longitude"] + 180) % 360 - 180