From f83c5a585e0844e9f0a6035131f8b4127d904588 Mon Sep 17 00:00:00 2001
From: Meng Li <34143965+Meng6@users.noreply.github.com>
Date: Mon, 1 Feb 2021 13:57:12 -0500
Subject: [PATCH] Fix HR intraday bug: minutesonZONE features are 0

---
 src/features/fitbit_heartrate_intraday/rapids/main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/features/fitbit_heartrate_intraday/rapids/main.py b/src/features/fitbit_heartrate_intraday/rapids/main.py
index 1a853722..d78b327d 100644
--- a/src/features/fitbit_heartrate_intraday/rapids/main.py
+++ b/src/features/fitbit_heartrate_intraday/rapids/main.py
@@ -55,10 +55,11 @@ def extractHRFeaturesFromIntradayData(heartrate_intraday_data, features, time_se
 
             # get number of minutes in each heart rate zone
             for feature_name in list(set(["minutesonoutofrangezone", "minutesonfatburnzone", "minutesoncardiozone", "minutesonpeakzone"]) & set(features)):
-                heartrate_zone = heartrate_intraday_data[heartrate_intraday_data["heartrate_zone"] == feature_name[17:-4]]
+                heartrate_zone = heartrate_intraday_data[heartrate_intraday_data["heartrate_zone"] == feature_name[9:-4]]
                 heartrate_intraday_features[feature_name] = heartrate_zone.groupby(["local_segment"])["device_id"].count() / num_rows_per_minute
                 heartrate_intraday_features.fillna(value={feature_name: 0}, inplace=True)
-        heartrate_intraday_features.reset_index(inplace=True)
+            
+            heartrate_intraday_features.reset_index(inplace=True)
 
     return heartrate_intraday_features