From daa9708e332299fc3c4e459360fac42894e1c4aa Mon Sep 17 00:00:00 2001 From: NicholasTurner23 Date: Mon, 2 Dec 2024 13:29:44 +0300 Subject: [PATCH] Switch extract device data method with new implementation. --- src/workflows/airqo_etl_utils/airqo_utils.py | 4 ++-- src/workflows/dags/airqo_measurements.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/workflows/airqo_etl_utils/airqo_utils.py b/src/workflows/airqo_etl_utils/airqo_utils.py index 57d9cec453..098b058996 100644 --- a/src/workflows/airqo_etl_utils/airqo_utils.py +++ b/src/workflows/airqo_etl_utils/airqo_utils.py @@ -395,7 +395,7 @@ def restructure_airqo_mobile_data_for_bigquery(data: pd.DataFrame) -> pd.DataFra return Utils.populate_missing_columns(data=data, columns=cols) @staticmethod - def extract_devices_data( + def extract_devices_data_( start_date_time: str, end_date_time: str, device_category: DeviceCategory, @@ -536,7 +536,7 @@ def extract_devices_data( return devices_data @staticmethod - def extract_devices_data_( + def extract_devices_data( start_date_time: str, end_date_time: str, device_category: DeviceCategory, diff --git a/src/workflows/dags/airqo_measurements.py b/src/workflows/dags/airqo_measurements.py index 7bc76e0ce9..392599e299 100644 --- a/src/workflows/dags/airqo_measurements.py +++ b/src/workflows/dags/airqo_measurements.py @@ -493,7 +493,7 @@ def extract_raw_data(**kwargs): start_date_time = date_to_str_hours(hour_of_day) end_date_time = datetime.strftime(hour_of_day, "%Y-%m-%dT%H:59:59Z") - return AirQoDataUtils.extract_devices_data_( + return AirQoDataUtils.extract_devices_data( start_date_time=start_date_time, end_date_time=end_date_time, device_category=DeviceCategory.LOW_COST,