Just some cleanup

airqo-platform · Dec 13, 2024 · 5a2ac8d · 5a2ac8d
1 parent 8c75b3d
commit 5a2ac8d
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,
                 )