fix layer id for names

MIT-SPARK · Feb 8, 2025 · 892e10c · 892e10c
1 parent e0fcd8c
commit 892e10c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/src/spark_dsg/open3d_visualization.py b/python/src/spark_dsg/open3d_visualization.py
@@ -63,7 +63,7 @@
     DsgLayers.name_to_layer_id(DsgLayers.OBJECTS),
     DsgLayers.name_to_layer_id(DsgLayers.PLACES),
     DsgLayers.name_to_layer_id(DsgLayers.ROOMS),
-    DsgLayers.name_to_layer_id(DsgLayers.BUILDINGS)
+    DsgLayers.name_to_layer_id(DsgLayers.BUILDINGS),
 )
 
 CATEGORY_MAP = {
@@ -340,7 +340,9 @@ def _update_names(self, G):
         self._names = [None] * len(self._id_map)
         for node_id in self._id_map:
             node = G.get_node(node_id)
-            if node.layer == DsgLayers.OBJECTS or node.layer == DsgLayers.ROOMS:
+            if node.layer == DsgLayers.name_to_layer_id(
+                DsgLayers.OBJECTS
+            ) or node.layer == DsgLayers.name_to_layer_id(DsgLayers.ROOMS):
                 self._names[self._id_map[node_id]] = node.attributes.name
 
         for name, point in zip(self._names, self._points):