[mypy] Enable type checking for test directory (#5017)

This commit is contained in:
Cyrus Leung
2024-06-15 12:45:31 +08:00
committed by GitHub
parent 1b8a0d71cf
commit 0e9164b40a
92 changed files with 509 additions and 378 deletions

View File

@@ -2,6 +2,7 @@ import argparse
import asyncio
import sys
from io import StringIO
from typing import Awaitable, List
import aiohttp
@@ -114,7 +115,7 @@ async def main(args):
)
# Submit all requests in the file to the engine "concurrently".
response_futures = []
response_futures: List[Awaitable[BatchRequestOutput]] = []
for request_json in (await read_file(args.input_file)).strip().split("\n"):
request = BatchRequestInput.model_validate_json(request_json)
response_futures.append(run_request(openai_serving_chat, request))