Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
Varun Sundar Rabindranath committed Sep 27, 2024
1 parent 931a4ac commit 7f047fe
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -529,6 +529,7 @@ def get_client_text_generations(completions: List[Completion]) -> List[str]:
assert all([len(x.choices) == 1 for x in completions])
return [x.choices[0].text for x in completions]


def get_client_text_logprob_generations(
completions: List[Completion]) -> List[TextTextLogprobs]:
'''Operates on the output of a request made to an Open-AI-protocol
Expand Down

0 comments on commit 7f047fe

Please sign in to comment.