From 96c4990165f8096da3de30954811b42b731a286d Mon Sep 17 00:00:00 2001 From: Gunnlaugur Thor Briem Date: Tue, 3 Mar 2020 22:57:05 +0000 Subject: [PATCH] fix unused imports and style --- src/transformers/modeling_tf_distilbert.py | 2 +- src/transformers/modeling_tf_openai.py | 1 - src/transformers/modeling_tf_roberta.py | 3 +-- src/transformers/modeling_tf_t5.py | 2 +- src/transformers/modeling_tf_xlm.py | 9 +-------- tests/test_modeling_tf_common.py | 5 +++-- 6 files changed, 7 insertions(+), 15 deletions(-) diff --git a/src/transformers/modeling_tf_distilbert.py b/src/transformers/modeling_tf_distilbert.py index 64cd93863a..6f6eaa3be0 100644 --- a/src/transformers/modeling_tf_distilbert.py +++ b/src/transformers/modeling_tf_distilbert.py @@ -24,7 +24,7 @@ import tensorflow as tf from .configuration_distilbert import DistilBertConfig from .file_utils import add_start_docstrings, add_start_docstrings_to_callable -from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, get_initializer, keras_serializable, shape_list +from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, get_initializer, shape_list logger = logging.getLogger(__name__) diff --git a/src/transformers/modeling_tf_openai.py b/src/transformers/modeling_tf_openai.py index 50e0af3e6c..6a97ae7786 100644 --- a/src/transformers/modeling_tf_openai.py +++ b/src/transformers/modeling_tf_openai.py @@ -29,7 +29,6 @@ from .modeling_tf_utils import ( TFSequenceSummary, TFSharedEmbeddings, get_initializer, - keras_serializable, shape_list, ) diff --git a/src/transformers/modeling_tf_roberta.py b/src/transformers/modeling_tf_roberta.py index 3347d7e041..31fb43f1cc 100644 --- a/src/transformers/modeling_tf_roberta.py +++ b/src/transformers/modeling_tf_roberta.py @@ -20,11 +20,10 @@ import logging import tensorflow as tf -from . import PretrainedConfig from .configuration_roberta import RobertaConfig from .file_utils import add_start_docstrings, add_start_docstrings_to_callable from .modeling_tf_bert import TFBertEmbeddings, TFBertMainLayer, gelu -from .modeling_tf_utils import TFPreTrainedModel, get_initializer, keras_serializable, shape_list +from .modeling_tf_utils import TFPreTrainedModel, get_initializer, shape_list logger = logging.getLogger(__name__) diff --git a/src/transformers/modeling_tf_t5.py b/src/transformers/modeling_tf_t5.py index f233074895..41520e044b 100644 --- a/src/transformers/modeling_tf_t5.py +++ b/src/transformers/modeling_tf_t5.py @@ -25,7 +25,7 @@ import tensorflow as tf from .configuration_t5 import T5Config from .file_utils import DUMMY_INPUTS, DUMMY_MASK, add_start_docstrings -from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, keras_serializable, shape_list +from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, shape_list logger = logging.getLogger(__name__) diff --git a/src/transformers/modeling_tf_xlm.py b/src/transformers/modeling_tf_xlm.py index 8a3f200e6f..6e94a7206e 100644 --- a/src/transformers/modeling_tf_xlm.py +++ b/src/transformers/modeling_tf_xlm.py @@ -25,14 +25,7 @@ import tensorflow as tf from .configuration_xlm import XLMConfig from .file_utils import add_start_docstrings, add_start_docstrings_to_callable -from .modeling_tf_utils import ( - TFPreTrainedModel, - TFSequenceSummary, - TFSharedEmbeddings, - get_initializer, - keras_serializable, - shape_list, -) +from .modeling_tf_utils import TFPreTrainedModel, TFSequenceSummary, TFSharedEmbeddings, get_initializer, shape_list logger = logging.getLogger(__name__) diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index 8f4cae7491..7c93ea16cd 100644 --- a/tests/test_modeling_tf_common.py +++ b/tests/test_modeling_tf_common.py @@ -102,8 +102,9 @@ class TFModelTesterMixin: for module_member_name in dir(module) if module_member_name.endswith("MainLayer") for module_member in (getattr(module, module_member_name),) - if isinstance(module_member, type) and tf.keras.layers.Layer in module_member.__bases__ - and getattr(module_member, '_keras_serializable', False) + if isinstance(module_member, type) + and tf.keras.layers.Layer in module_member.__bases__ + and getattr(module_member, "_keras_serializable", False) ) for main_layer_class in tf_main_layer_classes: main_layer = main_layer_class(config)