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]

https://github.com/waleedka/hiddenlayer/blob/45243d51fd78cb6edc45cca50d29b04fb4b35511/hiddenlayer/graph.py#L359 replace this line with return self.build_dot()._repr_image_svg_xml()