Skip to content

Fix label name in DataCollatorForNextSentencePrediction test#8048

Merged
LysandreJik merged 1 commit intomasterfrom
fix_8034
Oct 26, 2020
Merged

Fix label name in DataCollatorForNextSentencePrediction test#8048
LysandreJik merged 1 commit intomasterfrom
fix_8034

Conversation

@sgugger
Copy link
Copy Markdown
Collaborator

@sgugger sgugger commented Oct 26, 2020

What does this PR do?

Labels have been renamed in DataCollatorForNextSentencePrediction to go with the fact masked_lm_labels is a deprecated argument, but the corresponding test was not adjusted accordingly, this PR fixes that.

Fixes #8034

Copy link
Copy Markdown
Member

@LysandreJik LysandreJik left a comment

Choose a reason for hiding this comment

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

LGTM!

@LysandreJik LysandreJik merged commit 0774786 into master Oct 26, 2020
@LysandreJik LysandreJik deleted the fix_8034 branch October 26, 2020 13:23
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.

DataCollatorIntegrationTest::test_nsp failing on GPU

2 participants