Use walrus for unit

home-assistant · Sep 13, 2021 · 619dc7d · 619dc7d
1 parent 6641e3f
commit 619dc7d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeassistant/components/zwave_js/migrate.py b/homeassistant/components/zwave_js/migrate.py
@@ -207,8 +207,8 @@ def add_entity_value(
         assert device_entry
 
         # Normalize unit of measurement.
-        unit = entity_entry.unit_of_measurement
-        unit = unit and unit.lower()
+        if unit := entity_entry.unit_of_measurement:
+            unit = unit.lower()
         if unit == "":
             unit = None