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

fix unused imports and style

Browse Source
This commit is contained in:
Gunnlaugur Thor Briem
2020-03-03 22:57:05 +00:00
parent 470753bcf5
commit 96c4990165
6 changed files with 7 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_tf_distilbert.py
View File

@@ -24,7 +24,7 @@ import tensorflow as tf
from .configuration_distilbert import DistilBertConfig from .configuration_distilbert import DistilBertConfig
from .file_utils import add_start_docstrings, add_start_docstrings_to_callable 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__) logger = logging.getLogger(__name__)

1
src/transformers/modeling_tf_openai.py
View File

@@ -29,7 +29,6 @@ from .modeling_tf_utils import (
TFSequenceSummary, TFSequenceSummary,
TFSharedEmbeddings, TFSharedEmbeddings,
get_initializer, get_initializer,
keras_serializable,
shape_list, shape_list,
) )

3
src/transformers/modeling_tf_roberta.py
View File

@@ -20,11 +20,10 @@ import logging
import tensorflow as tf import tensorflow as tf
from . import PretrainedConfig
from .configuration_roberta import RobertaConfig from .configuration_roberta import RobertaConfig
from .file_utils import add_start_docstrings, add_start_docstrings_to_callable from .file_utils import add_start_docstrings, add_start_docstrings_to_callable
from .modeling_tf_bert import TFBertEmbeddings, TFBertMainLayer, gelu 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__) logger = logging.getLogger(__name__)

2
src/transformers/modeling_tf_t5.py
View File

@@ -25,7 +25,7 @@ import tensorflow as tf
from .configuration_t5 import T5Config from .configuration_t5 import T5Config
from .file_utils import DUMMY_INPUTS, DUMMY_MASK, add_start_docstrings 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__) logger = logging.getLogger(__name__)

9
src/transformers/modeling_tf_xlm.py
View File

@@ -25,14 +25,7 @@ import tensorflow as tf
from .configuration_xlm import XLMConfig from .configuration_xlm import XLMConfig
from .file_utils import add_start_docstrings, add_start_docstrings_to_callable from .file_utils import add_start_docstrings, add_start_docstrings_to_callable
from .modeling_tf_utils import ( from .modeling_tf_utils import TFPreTrainedModel, TFSequenceSummary, TFSharedEmbeddings, get_initializer, shape_list
TFPreTrainedModel,
TFSequenceSummary,
TFSharedEmbeddings,
get_initializer,
keras_serializable,
shape_list,
)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

5
tests/test_modeling_tf_common.py
View File

@@ -102,8 +102,9 @@ class TFModelTesterMixin:
for module_member_name in dir(module) for module_member_name in dir(module)
if module_member_name.endswith("MainLayer") if module_member_name.endswith("MainLayer")
for module_member in (getattr(module, module_member_name),) for module_member in (getattr(module, module_member_name),)
if isinstance(module_member, type) and tf.keras.layers.Layer in module_member.__bases__ if isinstance(module_member, type)
and getattr(module_member, '_keras_serializable', False) 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: for main_layer_class in tf_main_layer_classes:
main_layer = main_layer_class(config) main_layer = main_layer_class(config)