hf3

nokia · Nov 6, 2024 · 1797eb4 · 1797eb4
1 parent 019748b
commit 1797eb4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/moler/device/proxy_pc.py b/moler/device/proxy_pc.py
@@ -155,7 +155,8 @@ def _prepare_state_prompts(self):
             else:
                 state_prompts = self._prepare_state_prompts_without_proxy_pc()
         except KeyError as ke:
-            raise DeviceFailure(f"Wrong configuration. Cannot get prompts. {ke} {repr(ke)}")
+            raise DeviceFailure(device=self.__class__.__name__,
+                message=f"Wrong configuration. Cannot get prompts. {ke} {repr(ke)}")
 
         self._update_dict(self._state_prompts, state_prompts)
 

diff --git a/moler/device/unixremote.py b/moler/device/unixremote.py
@@ -451,7 +451,8 @@ def _overwrite_prompts(self):
                     self._configurations[UnixRemote.connection_hops][UnixRemote.unix_local][UnixRemote.unix_remote][
                         "command_params"]["expected_prompt"]
         except KeyError as ke:
-            raise DeviceFailure(f"Wrong configuration. Cannot get prompts. {ke} {repr(ke)}")
+            raise DeviceFailure(device=self.__class__.__name__,
+                message=f"Wrong configuration. Cannot get prompts. {ke} {repr(ke)}")
 
     def _get_packages_for_state(self, state, observer):
         """