Fix broken graphviz import

joowani · Mar 24, 2022 · bd75150 · bd75150
1 parent c0586d8
commit bd75150
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/binarytree/__init__.py b/binarytree/__init__.py
@@ -21,7 +21,12 @@
 from typing import Any, Deque, Dict, Iterator, List, Optional, Tuple, Union
 
 from graphviz import Digraph, nohtml
-from graphviz.exceptions import ExecutableNotFound
+
+try:
+    from graphviz.exceptions import ExecutableNotFound
+except ImportError:
+    # noinspection PyProtectedMember
+    from graphviz import ExecutableNotFound
 from pkg_resources import get_distribution
 
 from binarytree.exceptions import (