Fix tests imports dpr (#5576)

* fix test imports * fix max_length * style * fix tests
2020-07-07 16:35:12 +02:00
parent d4886173b2
commit 4fedc1256c
3 changed files with 31 additions and 9 deletions
--- a/tests/test_modeling_dpr.py
+++ b/tests/test_modeling_dpr.py
@@ -17,10 +17,10 @@
 import unittest

 from transformers import is_torch_available
+from transformers.testing_utils import require_torch, slow, torch_device

 from .test_configuration_common import ConfigTester
 from .test_modeling_common import ModelTesterMixin, ids_tensor
-from .utils import require_torch, slow, torch_device


 if is_torch_available():