fix: Change time_period_str to parse_duration

jcgoette · Oct 19, 2022 · b2f2ebb · b2f2ebb
1 parent c3d644d
commit b2f2ebb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/babybuddy/const.py b/custom_components/babybuddy/const.py
@@ -5,6 +5,7 @@
 from dataclasses import dataclass, field
 from typing import Final
 
+import homeassistant.util.dt as dt_util
 from homeassistant.components.select import SelectEntityDescription
 from homeassistant.components.sensor import (
     SensorDeviceClass,
@@ -13,7 +14,6 @@
 )
 from homeassistant.components.switch import SwitchEntityDescription
 from homeassistant.const import ATTR_TIME, TIME_MINUTES
-from homeassistant.helpers.config_validation import time_period_str
 
 DOMAIN: Final = "babybuddy"
 
@@ -138,7 +138,7 @@ class BabyBuddyEntityDescription(SensorEntityDescription, SwitchEntityDescriptio
         native_unit_of_measurement=TIME_MINUTES,
         state_class=SensorStateClass.MEASUREMENT,
         state_key=lambda value: int(
-            time_period_str(value[ATTR_DURATION]).total_seconds() / 60
+            dt_util.parse_duration(value[ATTR_DURATION]).total_seconds() / 60
         ),
     ),
     BabyBuddyEntityDescription(
@@ -160,7 +160,7 @@ class BabyBuddyEntityDescription(SensorEntityDescription, SwitchEntityDescriptio
         native_unit_of_measurement=TIME_MINUTES,
         state_class=SensorStateClass.MEASUREMENT,
         state_key=lambda value: int(
-            time_period_str(value[ATTR_DURATION]).total_seconds() / 60
+            dt_util.parse_duration(value[ATTR_DURATION]).total_seconds() / 60
         ),
     ),
     BabyBuddyEntityDescription(