Skip to content

Commit c9ea6d3

Browse files
committed
add explicit typing for messages
1 parent c495fc2 commit c9ea6d3

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

airbyte_cdk/test/entrypoint_wrapper.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ def _run_command(
319319
parsed_args = AirbyteEntrypoint.parse_args(args)
320320

321321
source_entrypoint = AirbyteEntrypoint(source)
322-
messages = []
322+
messages: list[str] = []
323323
uncaught_exception = None
324324
try:
325325
for message in source_entrypoint.run(parsed_args):
@@ -334,8 +334,10 @@ def _run_command(
334334
captured_logs = log_capture_buffer.getvalue().split("\n")[:-1]
335335

336336
parent_logger.removeHandler(stream_handler)
337-
338-
return EntrypointOutput(messages + captured_logs, uncaught_exception=uncaught_exception)
337+
return EntrypointOutput(
338+
messages=messages + captured_logs,
339+
uncaught_exception=uncaught_exception,
340+
)
339341

340342

341343
def discover(

0 commit comments

Comments
 (0)