Skip to content

Commit 5ffc20f

Browse files
dicksontsairushilpatel0
authored andcommitted
Fix lint
Signed-off-by: Rushil Patel <[email protected]>
1 parent 8ea3a12 commit 5ffc20f

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

tests/test_streaming_client.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -403,30 +403,26 @@ async def message_stream():
403403
# Output a valid result
404404
print('{"type": "result", "subtype": "success", "duration_ms": 100, "duration_api_ms": 50, "is_error": false, "num_turns": 1, "session_id": "test", "total_cost_usd": 0.001}')
405405
""")
406-
406+
407407
Path(test_script).chmod(0o755)
408408

409409
try:
410410
# Mock _find_cli to return python executing our test script
411411
with patch.object(
412-
SubprocessCLITransport,
413-
"_find_cli",
414-
return_value=sys.executable
412+
SubprocessCLITransport, "_find_cli", return_value=sys.executable
415413
):
416414
# Mock _build_command to add our test script as first argument
417415
original_build_command = SubprocessCLITransport._build_command
418-
416+
419417
def mock_build_command(self):
420418
# Get original command
421419
cmd = original_build_command(self)
422420
# Replace the CLI path with python + script
423421
cmd[0] = test_script
424422
return cmd
425-
423+
426424
with patch.object(
427-
SubprocessCLITransport,
428-
"_build_command",
429-
mock_build_command
425+
SubprocessCLITransport, "_build_command", mock_build_command
430426
):
431427
# Run query with async iterable
432428
messages = []

0 commit comments

Comments
 (0)