Update vyos_ssh.py to enter and exit config mode on save (#3296)

ktbyers · Oct 17, 2023 · e2135e7 · e2135e7
1 parent 11c80ea
commit e2135e7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/netmiko/vyos/vyos_ssh.py b/netmiko/vyos/vyos_ssh.py
@@ -136,9 +136,11 @@ def save_config(
         self, cmd: str = "save", confirm: bool = False, confirm_response: str = ""
     ) -> str:
         """Saves Config."""
-        output = super().save_config(
+        output = self.config_mode()
+        output += super().save_config(
             cmd=cmd, confirm=confirm, confirm_response=confirm_response
         )
+        output += self.exit_config_mode()
         if "Done" not in output:
             raise ValueError(f"Save failed with following errors:\n\n{output}")
         return output