Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix Batcher iterator break condition #2655

Merged
merged 1 commit into from
Dec 24, 2024
Merged

Conversation

hhllhhyyds
Copy link
Contributor

Fix Batcher iterator break when return_last_incomplete_batch and items.is_empty (#2654)

@hhllhhyyds hhllhhyyds changed the title Fix Batcher iterator break condition (#2654) Fix Batcher iterator break condition Dec 1, 2024
Copy link
Contributor

@nerdai nerdai left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I just encountered this bug as well. The fix looks good to me.

Do we have any unit tests for Batcher yet?

@LaurentMazare LaurentMazare merged commit 11aa30b into huggingface:main Dec 24, 2024
10 checks passed
@LaurentMazare
Copy link
Collaborator

Let's merge this for now but feel free to add a test if you want.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants