add multiple imports

ekrichbaum · Dec 15, 2017 · b4a1d4f · b4a1d4f
1 parent cab7340
commit b4a1d4f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/napalm_panos/panos.py b/napalm_panos/panos.py
@@ -30,7 +30,10 @@
 from napalm_base.exceptions import ConnectionException, ReplaceConfigException,\
                                    MergeConfigException
 
-from napalm_base.base import NetworkDriver
+try:
+    from napalm_base import NetworkDriver
+except ImportError:
+    from napalm_base.base import NetworkDriver
 
 from napalm_base.utils import py23_compat