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.

3 participants