fix: The 'warn' method is deprecated (#11105)
* The 'warn' method is deprecated * fix test
This commit is contained in:
@@ -353,7 +353,7 @@ def main():
|
||||
if list(sorted(label_name_to_id.keys())) == list(sorted(label_list)):
|
||||
label_to_id = {i: int(label_name_to_id[label_list[i]]) for i in range(num_labels)}
|
||||
else:
|
||||
logger.warn(
|
||||
logger.warning(
|
||||
"Your model seems to have been trained with labels, but they don't match the dataset: ",
|
||||
f"model labels: {list(sorted(label_name_to_id.keys()))}, dataset labels: {list(sorted(label_list))}."
|
||||
"\nIgnoring the model labels as a result.",
|
||||
@@ -362,7 +362,7 @@ def main():
|
||||
label_to_id = {v: i for i, v in enumerate(label_list)}
|
||||
|
||||
if data_args.max_seq_length > tokenizer.model_max_length:
|
||||
logger.warn(
|
||||
logger.warning(
|
||||
f"The max_seq_length passed ({data_args.max_seq_length}) is larger than the maximum length for the"
|
||||
f"model ({tokenizer.model_max_length}). Using max_seq_length={tokenizer.model_max_length}."
|
||||
)
|
||||
|
||||
@@ -51,7 +51,7 @@ class HfArgumentParserTest(unittest.TestCase):
|
||||
# should be able to log warnings (if default settings weren't overridden by `pytest --log-level-all`)
|
||||
if level_origin <= logging.WARNING:
|
||||
with CaptureLogger(logger) as cl:
|
||||
logger.warn(msg)
|
||||
logger.warning(msg)
|
||||
self.assertEqual(cl.out, msg + "\n")
|
||||
|
||||
# this is setting the level for all of `transformers.*` loggers
|
||||
@@ -59,7 +59,7 @@ class HfArgumentParserTest(unittest.TestCase):
|
||||
|
||||
# should not be able to log warnings
|
||||
with CaptureLogger(logger) as cl:
|
||||
logger.warn(msg)
|
||||
logger.warning(msg)
|
||||
self.assertEqual(cl.out, "")
|
||||
|
||||
# should be able to log warnings again
|
||||
|
||||
@@ -234,7 +234,7 @@ class TrainerCallbackTest(unittest.TestCase):
|
||||
self.assertEqual(events, self.get_expected_events(trainer))
|
||||
|
||||
# warning should be emitted for duplicated callbacks
|
||||
with unittest.mock.patch("transformers.trainer_callback.logger.warn") as warn_mock:
|
||||
with unittest.mock.patch("transformers.trainer_callback.logger.warning") as warn_mock:
|
||||
trainer = self.get_trainer(
|
||||
callbacks=[MyTestTrainerCallback, MyTestTrainerCallback],
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user