Merge pull request #4068 from NicholasTurner23/update-fix/Clean_up

Update fix/clean up
airqo-platform · Dec 13, 2024 · 549b847 · 549b847
2 parents d8cdae1 + 44abcbc
commit 549b847
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/analytics/api/views/data.py b/src/analytics/api/views/data.py
@@ -163,11 +163,14 @@ def post(self):
                 )
             if minimum_output:
                 # Drop unnecessary columns
+                columns_to_drop = ["site_id"]
+                columns_to_drop.append("timestamp") if frequency in [
+                    "hourly",
+                    "daily",
+                ] else columns_to_drop
+                print(data_frame.columns.to_list())
                 data_frame.drop(
-                    columns=[
-                        "site_id",
-                        "timestamp",
-                    ],
+                    columns=columns_to_drop,
                     inplace=True,
                 )