AttributeError: 'Digraph' object has no attribute '_repr_svg_'

[Uday-Shankar-Shanthamallu]

I am following the demo notebook for pytorch (version 1.9.1) and I get the following error:

model = torchvision.models.alexnet()

# Build HiddenLayer graph
hl_graph = hl.build_graph(model, torch.zeros([1, 3, 224, 224]))

# Use a different color theme
hl_graph.theme = hl.graph.THEMES["blue"].copy()  # Two options: basic and blue
hl_graph

---

AttributeError Traceback (most recent call last)
~/opt/anaconda3/envs/gexp/lib/python3.8/site-packages/IPython/core/formatters.py in call(self, obj)
343 method = get_real_method(obj, self.print_method)
344 if method is not None:
--> 345 return method()
346 return None
347 else:

~/opt/anaconda3/envs/gexp/lib/python3.8/site-packages/hiddenlayer/graph.py in repr_svg(self)
357 def repr_svg(self):
358 """Allows Jupyter notebook to render the graph automatically."""
--> 359 return self.build_dot().repr_svg()
360
361 def save(self, path, format="pdf"):

AttributeError: 'Digraph' object has no attribute 'repr_svg'

[lpacit]

Same issue here, did you manage to solve it?

[pyjhzwh]

hiddenlayer/hiddenlayer/graph.py

Line 359 in 45243d5

return self.build_dot()._repr_svg_()

replace this line with
return self.build_dot()._repr_image_svg_xml()