Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

[Rerank] Ensure other batch fields are in re-ranker batch_act #4296

Merged
merged 1 commit into from Jan 7, 2022

Conversation

klshuster
Copy link
Contributor

Patch description
The AbstractGeneratorRerankAgentMixin was ignoring anything besides beam_texts and text within a batch reply, resulting in failures to bubble up e.g. agent metrics. This PR fixes that.

Testing steps
Tested locally, also ensured that pytest test_light_whoami.py passed.

@jaseweston jaseweston merged commit 9e54cc9 into main Jan 7, 2022
@jaseweston jaseweston deleted the update_reranker branch January 7, 2022 21:29
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants