Fix export of color in material that is not child of robot tag

fishbotics · Sep 26, 2024 · b4436aa · b4436aa
1 parent 0f774a7
commit b4436aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/urchin/urdf.py b/urchin/urdf.py
@@ -1054,10 +1054,13 @@ def _to_xml(self, parent, path):
                 color.attrib["rgba"] = np.array2string(self.color)[1:-1]
                 node.append(color)
 
-        # For non-top-level elements just save the material with a name
         else:
             node = ET.Element("material")
             node.attrib["name"] = self.name
+            if self.color is not None:
+                color = ET.Element("color")
+                color.attrib["rgba"] = np.array2string(self.color)[1:-1]
+                node.append(color)
         return node
 
     def copy(self, prefix="", scale=None):