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 2 tests in FillMaskPipelineTests #27889

Merged
merged 5 commits into from
Dec 8, 2023
Merged

Fix 2 tests in FillMaskPipelineTests #27889

merged 5 commits into from
Dec 8, 2023

Conversation

ydshieh
Copy link
Collaborator

@ydshieh ydshieh commented Dec 7, 2023

What does this PR do?

The test is introduced in #26234 which doesn't work as expected.

This PR update the values.

@ydshieh ydshieh marked this pull request as draft December 7, 2023 12:25
@ydshieh ydshieh marked this pull request as ready for review December 7, 2023 12:41
Comment on lines 225 to +230
self.assertEqual(
nested_simplify(outputs, decimals=6),
[{"sequence": x["sequence"][:100]} for x in simplified],
[
{"sequence": "My name is grouped", "score": 2.2e-05, "token": 38015, "token_str": " grouped"},
{"sequence": "My name is accuser", "score": 2.1e-05, "token": 25506, "token_str": " accuser"},
{"sequence": f"My name is,{dummy_str}"[:100]},
{"sequence": f"My name is:,{dummy_str}"[:100]},
],
Copy link
Collaborator Author

Choose a reason for hiding this comment

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

The input ids are truncated in this (part of) test. The output is also truncated. Let's just compare the first 100 characters.

Comment on lines +235 to +236
{"score": 0.2819, "token": 6, "token_str": ","},
{"score": 0.0954, "token": 46686, "token_str": ":,"},
Copy link
Collaborator Author

Choose a reason for hiding this comment

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

just update some values (contributor might have different env.)

@ydshieh ydshieh requested a review from ArthurZucker December 7, 2023 12:44
@ydshieh ydshieh merged commit e366937 into main Dec 8, 2023
18 checks passed
@ydshieh ydshieh deleted the fix_fill_mask_pipe_test branch December 8, 2023 13:55
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.

2 participants