@@ -260,22 +260,22 @@ def write_png(self, filename, prog=None): # pragma: no cover
260
260
261
261
for frame in self .nodes :
262
262
node = pydot .Node (
263
- _dot_display_name (frame ), style = "filled" ,
263
+ _dot_display_name (str ( frame ) ), style = "filled" ,
264
264
fillcolor = frame_color , shape = "egg" )
265
265
graph .add_node (node )
266
266
for frames , A2B in self ._transforms .items ():
267
267
frame_a , frame_b = frames
268
268
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 )))
271
271
node = pydot .Node (
272
272
connection_name , style = "filled" , fillcolor = connection_color ,
273
273
shape = "note" )
274
274
graph .add_node (node )
275
- a_name = _dot_display_name (frame_a )
275
+ a_name = _dot_display_name (str ( frame_a ) )
276
276
a_edge = pydot .Edge (connection_name , a_name , penwidth = 3 )
277
277
graph .add_edge (a_edge )
278
- b_name = _dot_display_name (frame_b )
278
+ b_name = _dot_display_name (str ( frame_b ) )
279
279
b_edge = pydot .Edge (connection_name , b_name , penwidth = 3 )
280
280
graph .add_edge (b_edge )
281
281
0 commit comments