Import pydot first before trying backups (#20682)

keras-team · Dec 23, 2024 · c1316e5 · c1316e5
1 parent 3dd958b
commit c1316e5
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/keras/src/utils/model_visualization.py b/keras/src/utils/model_visualization.py
@@ -8,16 +8,15 @@
 from keras.src.utils import io_utils
 
 try:
-    # pydot-ng is a fork of pydot that is better maintained.
-    import pydot_ng as pydot
+    import pydot
 except ImportError:
-    # pydotplus is an improved version of pydot
+    # pydot_ng and pydotplus are older forks of pydot
+    # which may still be used by some users
     try:
-        import pydotplus as pydot
+        import pydot_ng as pydot
     except ImportError:
-        # Fall back on pydot if necessary.
         try:
-            import pydot
+            import pydotplus as pydot
         except ImportError:
             pydot = None