Account for deprecation of DiGraph._repr_svg_ in graphviz

joowani · Jun 28, 2022 · 87ff18a · 87ff18a
1 parent 397e65f
commit 87ff18a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/binarytree/__init__.py b/binarytree/__init__.py
@@ -508,8 +508,12 @@ def _repr_svg_(self) -> str:  # pragma: no cover
         .. _Jupyter notebooks: https://jupyter.org
         """
         try:
-            # noinspection PyProtectedMember
-            return str(self.graphviz()._repr_svg_())
+            try:
+                # noinspection PyProtectedMember
+                return str(self.graphviz()._repr_svg_())
+            except AttributeError:
+                # noinspection PyProtectedMember
+                return self.graphviz()._repr_image_svg_xml()
 
         except (SubprocessError, ExecutableNotFound, FileNotFoundError):
             return self.svg()