Update src/transformers/data/data_collator.py

Rocketknight1 · Rocketknight1 · commit 959fc06757ed · 2025-11-11T14:10:48.000Z
diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py
@@ -725,7 +725,6 @@ def __post_init__(self):
                 warnings.warn(
                     "Random token replacement is not supported with whole word masking. "
                     "Setting mask_replace_prob to 1.",
-                    UserWarning,
                 )
                 self.mask_replace_prob = 1
                 self.random_replace_prob = 0

Original file line number	Diff line number	Diff line change
`@@ -725,7 +725,6 @@ def __post_init__(self):`
`725`	`725`	`warnings.warn(`
`726`	`726`	`"Random token replacement is not supported with whole word masking. "`
`727`	`727`	`"Setting mask_replace_prob to 1.",`
`728`		`- UserWarning,`
`729`	`728`	`)`
`730`	`729`	`self.mask_replace_prob = 1`
`731`	`730`	`self.random_replace_prob = 0`