fix: The 'warn' method is deprecated (#11105)

* The 'warn' method is deprecated * fix test
2021-04-07 06:20:06 -07:00
parent 247bed3857
commit c9035e4537
52 changed files with 68 additions and 66 deletions
--- a/tests/test_trainer_callback.py
+++ b/tests/test_trainer_callback.py
@@ -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],
            )