Skip to content

Make train_dataset attribute in _get_train_sampler optional #38226

Merged
ArthurZucker merged 2 commits into
mainfrom
fix-bc-sampler
May 20, 2025
Merged

Make train_dataset attribute in _get_train_sampler optional #38226
ArthurZucker merged 2 commits into
mainfrom
fix-bc-sampler

Conversation

@SunMarc

@SunMarc SunMarc commented May 20, 2025

Copy link
Copy Markdown
Member

What does this PR do?

This PR fixes a breaking change that was caused by #38090. We removed in that PR train_dataset arg but TRL lib uses it. We make it optional instead now

@SunMarc SunMarc changed the title Make train_dataset attribute in _get_train_sampler optional Make train_dataset attribute in _get_train_sampler optional May 20, 2025
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@SunMarc SunMarc requested a review from ArthurZucker May 20, 2025 12:12

@ArthurZucker ArthurZucker left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Thanks

@ArthurZucker ArthurZucker enabled auto-merge (squash) May 20, 2025 12:44
@ArthurZucker ArthurZucker merged commit bb3c642 into main May 20, 2025
21 checks passed
@ArthurZucker ArthurZucker deleted the fix-bc-sampler branch May 20, 2025 12:59
faaany pushed a commit to faaany/transformers that referenced this pull request May 21, 2025
xvyv99 pushed a commit to xvyv99/transformers that referenced this pull request May 21, 2025
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.

4 participants