diff --git a/scrapli/driver/generic/async_driver.py b/scrapli/driver/generic/async_driver.py index 84b8dcfd..9c98ea69 100644 --- a/scrapli/driver/generic/async_driver.py +++ b/scrapli/driver/generic/async_driver.py @@ -530,7 +530,13 @@ def callback_two(cls: GenericDriver, read_output: str): """ if initial_input is not None: self.channel.write(channel_input=f"{initial_input}{self.comms_return_char}") - return await self.read_callback(callbacks=callbacks, initial_input=None) + return await self.read_callback( + callbacks=callbacks, + initial_input=None, + read_output=read_output, + read_delay=read_delay, + read_timeout=read_timeout, + ) original_transport_timeout = self.timeout_transport diff --git a/scrapli/driver/generic/sync_driver.py b/scrapli/driver/generic/sync_driver.py index 6f855ef7..9b832e13 100644 --- a/scrapli/driver/generic/sync_driver.py +++ b/scrapli/driver/generic/sync_driver.py @@ -531,7 +531,13 @@ def callback_two(cls: GenericDriver, read_output: str): """ if initial_input is not None: self.channel.write(channel_input=f"{initial_input}{self.comms_return_char}") - return self.read_callback(callbacks=callbacks, initial_input=None) + return self.read_callback( + callbacks=callbacks, + initial_input=None, + read_output=read_output, + read_delay=read_delay, + read_timeout=read_timeout, + ) original_transport_timeout = self.timeout_transport