remove unused args from test generate

Signed-off-by: Travis Johnson <[email protected]>

Author: tjohnson31415

tests/e2e/test_spyre_async_llm.py

@@ -18,8 +18,6 @@ async def generate(
     output_kind: RequestOutputKind,
     max_tokens: int,
     n: int = 1,
-    prompt_logprobs: Optional[int] = None,
-    cancel_after: Optional[int] = None,
 ) -> tuple[int, str]:
     # Ensure generate doesn't complete too fast for cancellation test.
     await asyncio.sleep(0.2)
@@ -31,7 +29,6 @@ async def generate(
         output_kind=output_kind,
         seed=42,
         n=n,
-        prompt_logprobs=prompt_logprobs,
     )
     async for out in engine.generate(request_id=request_id,
                                      prompt=prompt,
@@ -43,9 +40,6 @@ async def generate(
         else:
             count = num_tokens
 
-        if cancel_after is not None and count >= cancel_after:
-            return count, request_id
-
         await asyncio.sleep(0.01)
 
     return count, request_id