feat: Rename control_unit device

alepee · Nov 18, 2024 · d82996a · d82996a
1 parent 74cef14
commit d82996a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/custom_components/hitachi_yutaki/__init__.py b/custom_components/hitachi_yutaki/__init__.py
@@ -63,12 +63,14 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     model_name = MODEL_NAMES.get(unit_model, "Unknown Model")
 
     # Add main unit device
+    device_name = DEVICE_CONTROL_UNIT.replace("_", " ").title()
+
     device_registry.async_get_or_create(
         config_entry_id=entry.entry_id,
         identifiers={(DOMAIN, f"{entry.entry_id}_{DEVICE_CONTROL_UNIT}")},
         manufacturer="Hitachi",
         model=model_name,
-        name=DEVICE_CONTROL_UNIT.title(),
+        name=device_name,
         via_device=(DOMAIN, f"{entry.entry_id}_{DEVICE_GATEWAY}"),
     )