Skip to content

Commit 9619e2a

Browse files
committed
Fix model unittest CI
1 parent 2e17e6f commit 9619e2a

File tree

3 files changed

+3
-12
lines changed

3 files changed

+3
-12
lines changed

tests/transformers/auto/test_configuration.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,7 @@
2525
from paddleformers.transformers.bert.configuration import BertConfig
2626
from paddleformers.utils.download import DownloadSource
2727
from paddleformers.utils.env import CONFIG_NAME
28-
from tests.testing_utils import (
29-
set_proxy,
30-
skip_for_none_ce_case,
31-
)
28+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
3229

3330
from ...utils.test_module.custom_configuration import CustomConfig
3431

tests/transformers/test_configuration_utils.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,7 @@
2727
from paddleformers.utils import CONFIG_NAME
2828
from paddleformers.utils.download import DownloadSource
2929
from paddleformers.utils.env import LEGACY_CONFIG_NAME
30-
from tests.testing_utils import (
31-
set_proxy,
32-
skip_for_none_ce_case,
33-
)
30+
from tests.testing_utils import set_proxy, skip_for_none_ce_case
3431

3532

3633
class FakeSimplePretrainedModelConfig(PretrainedConfig):

tests/transformers/test_shard_checkpoint.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,7 @@
3838
SAFE_WEIGHTS_NAME,
3939
)
4040
from paddleformers.utils.import_utils import is_paddle_cuda_available
41-
from tests.testing_utils import (
42-
require_package,
43-
skip_for_none_ce_case,
44-
)
41+
from tests.testing_utils import require_package, skip_for_none_ce_case
4542

4643

4744
class FakeConfig(PretrainedConfig):

0 commit comments

Comments
 (0)