From 5103ab5a2376ae099c22b728e9462e83be27724f Mon Sep 17 00:00:00 2001 From: Jared O'Connell Date: Thu, 15 Aug 2024 09:31:00 -0400 Subject: [PATCH] Fix linter errors --- src/arcaflow_plugin_sdk/atp.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/arcaflow_plugin_sdk/atp.py b/src/arcaflow_plugin_sdk/atp.py index 7e6d03f..540d9c4 100644 --- a/src/arcaflow_plugin_sdk/atp.py +++ b/src/arcaflow_plugin_sdk/atp.py @@ -114,7 +114,9 @@ def run_plugin( # potentially interfering with the atp pipes. original_stdout = sys.stdout original_stderr = sys.stderr - self.user_out_wrapper = io.TextIOWrapper(io.BytesIO(), sys.stdout.encoding) + self.user_out_wrapper = io.TextIOWrapper( + io.BytesIO(), sys.stdout.encoding + ) sys.stdout = self.user_out_wrapper sys.stderr = self.user_out_wrapper @@ -325,7 +327,7 @@ def start_step(self, run_id: str, step_id: str, config: typing.Any): ) self.user_out_wrapper.flush() - self.user_out_wrapper.seek(0) # go to start so that we can read stdout. + self.user_out_wrapper.seek(0) # go to start to read stdout. # Send WorkDoneMessage self.send_runtime_message( MessageType.WORK_DONE,