Skip to content

Commit 7c1318b

Browse files
Fix type in dot graph export
1 parent 1be863e commit 7c1318b

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pytransform3d/transform_manager/_transform_manager.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -260,22 +260,22 @@ def write_png(self, filename, prog=None): # pragma: no cover
260260

261261
for frame in self.nodes:
262262
node = pydot.Node(
263-
_dot_display_name(frame), style="filled",
263+
_dot_display_name(str(frame)), style="filled",
264264
fillcolor=frame_color, shape="egg")
265265
graph.add_node(node)
266266
for frames, A2B in self._transforms.items():
267267
frame_a, frame_b = frames
268268
connection_name = "%s to %s\n%s" % (
269-
_dot_display_name(frame_a), _dot_display_name(frame_b),
270-
str(np.round(A2B, 3)))
269+
_dot_display_name(str(frame_a)),
270+
_dot_display_name(str(frame_b)), str(np.round(A2B, 3)))
271271
node = pydot.Node(
272272
connection_name, style="filled", fillcolor=connection_color,
273273
shape="note")
274274
graph.add_node(node)
275-
a_name = _dot_display_name(frame_a)
275+
a_name = _dot_display_name(str(frame_a))
276276
a_edge = pydot.Edge(connection_name, a_name, penwidth=3)
277277
graph.add_edge(a_edge)
278-
b_name = _dot_display_name(frame_b)
278+
b_name = _dot_display_name(str(frame_b))
279279
b_edge = pydot.Edge(connection_name, b_name, penwidth=3)
280280
graph.add_edge(b_edge)
281281

0 commit comments

Comments
 (0)