From 83013ab7294fcb6db2a9595f626941c0e271ae2b Mon Sep 17 00:00:00 2001 From: Luke Lashley Date: Fri, 5 Dec 2025 16:57:24 -0500 Subject: [PATCH] Refactor room name retrieval in image.py --- custom_components/roborock_custom_map/image.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/custom_components/roborock_custom_map/image.py b/custom_components/roborock_custom_map/image.py index 03e481b..2928793 100644 --- a/custom_components/roborock_custom_map/image.py +++ b/custom_components/roborock_custom_map/image.py @@ -115,11 +115,11 @@ class RoborockMap(RoborockCoordinatedEntityV1, ImageEntity): return {} if map_data.rooms is not None: for room in map_data.rooms.values(): - name = map_data.rooms.get(room.number) + name = self._home_trait._rooms_trait.room_map.get(room.number) room.name = name.name if name else "Unknown" - + calibration = map_data.calibration() return { - "calibration_points": map_data.calibration(), + "calibration_points": calibration, "rooms": map_data.rooms, "zones": map_data.zones, }