fix tests

This commit is contained in:
Dakota 2025-10-29 10:55:10 -05:00
parent d5400460e8
commit c3a118f50d

View file

@ -151,7 +151,7 @@ class ServerHarness:
except KeyError as e:
raise KeyError(f"KeyError: {e} for key:\n{prompt}")
async def _tokens_and_logprobs_completion_wrapper(
async def tokens_and_logprobs_completion(
self, **kwargs
) -> tuple[list, list, list, list]:
"""