From ea8ffe36d3c97376f39a462820ad241c8dc52b95 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Thu, 12 Aug 2021 11:23:00 +0200 Subject: [PATCH] Proper import for unittest.mock.patch (#13085) --- tests/test_trainer_callback.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_trainer_callback.py b/tests/test_trainer_callback.py index 6ce90b8554..a7daee4fd0 100644 --- a/tests/test_trainer_callback.py +++ b/tests/test_trainer_callback.py @@ -15,6 +15,7 @@ import shutil import tempfile import unittest +from unittest.mock import patch from transformers import ( DefaultFlowCallback, @@ -234,7 +235,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.warning") as warn_mock: + with patch("transformers.trainer_callback.logger.warning") as warn_mock: trainer = self.get_trainer( callbacks=[MyTestTrainerCallback, MyTestTrainerCallback], )