SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Proper import for unittest.mock.patch (#13085)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-08-12 11:23:00 +02:00
committed by GitHub
parent d329b63369
commit ea8ffe36d3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_trainer_callback.py
View File

@@ -15,6 +15,7 @@
import shutil import shutil
import tempfile import tempfile
import unittest import unittest
from unittest.mock import patch
from transformers import ( from transformers import (
DefaultFlowCallback, DefaultFlowCallback,
@@ -234,7 +235,7 @@ class TrainerCallbackTest(unittest.TestCase):
self.assertEqual(events, self.get_expected_events(trainer)) self.assertEqual(events, self.get_expected_events(trainer))
# warning should be emitted for duplicated callbacks # 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( trainer = self.get_trainer(
callbacks=[MyTestTrainerCallback, MyTestTrainerCallback], callbacks=[MyTestTrainerCallback, MyTestTrainerCallback],
) )