From 294d84277d866c6e3720514d25e4a6af008ba490 Mon Sep 17 00:00:00 2001 From: Meng Li <34143965+Meng6@users.noreply.github.com> Date: Wed, 17 Mar 2021 15:21:31 -0400 Subject: [PATCH] Fix bug of sleep intraday PRICE provider when the dataframe of a segment is empty --- src/features/fitbit_sleep_intraday/price/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/features/fitbit_sleep_intraday/price/main.py b/src/features/fitbit_sleep_intraday/price/main.py index cda9f482..2d602305 100644 --- a/src/features/fitbit_sleep_intraday/price/main.py +++ b/src/features/fitbit_sleep_intraday/price/main.py @@ -181,12 +181,12 @@ def price_features(sensor_data_files, time_segment, provider, filter_data_by_seg features_fullnames = featuresFullNames(intraday_features_to_compute, sleep_levels_to_compute, day_types_to_compute) sleep_intraday_features = pd.DataFrame(columns=features_fullnames) - if sleep_intraday_data.empty: - return sleep_intraday_features - # Filter by segemnts and chunk episodes sleep_intraday_data = filter_data_by_segment(sleep_intraday_data, time_segment) + if sleep_intraday_data.empty: + return sleep_intraday_features + # Discard segments shorter than one day sleep_intraday_data["segment_length"] = (sleep_intraday_data["segment_end_timestamp"] - sleep_intraday_data["segment_start_timestamp"]) / 1000 # in seconds sleep_intraday_data = sleep_intraday_data[sleep_intraday_data["segment_length"] >= 24 * 60 * 60 - 1]