fix(eos): handle netmiko close (#1736)

Resolves #1735
napalm-automation · Aug 23, 2022 · d297056 · d297056
1 parent 029a38b
commit d297056
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/napalm/eos/eos.py b/napalm/eos/eos.py
@@ -36,7 +36,7 @@
 # third party libs
 import pyeapi
 from pyeapi.eapilib import ConnectionError, EapiConnection
-from netmiko import ConfigInvalidException
+from netmiko import BaseConnection, ConfigInvalidException
 
 # NAPALM base
 import napalm.base.helpers
@@ -231,7 +231,9 @@ def open(self):
     def close(self):
         """Implementation of NAPALM method close."""
         self.discard_config()
-        if hasattr(self.device.connection, "close") and callable(
+        if isinstance(self.device, BaseConnection):
+            self._netmiko_close()
+        elif hasattr(self.device.connection, "close") and callable(
             self.device.connection.close
         ):
             self.device.connection.close()