Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
96c49901
Commit
96c49901
authored
Mar 03, 2020
by
Gunnlaugur Thor Briem
Browse files
fix unused imports and style
parent
470753bc
Changes
6
Show whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
7 additions
and
15 deletions
+7
-15
src/transformers/modeling_tf_distilbert.py
src/transformers/modeling_tf_distilbert.py
+1
-1
src/transformers/modeling_tf_openai.py
src/transformers/modeling_tf_openai.py
+0
-1
src/transformers/modeling_tf_roberta.py
src/transformers/modeling_tf_roberta.py
+1
-2
src/transformers/modeling_tf_t5.py
src/transformers/modeling_tf_t5.py
+1
-1
src/transformers/modeling_tf_xlm.py
src/transformers/modeling_tf_xlm.py
+1
-8
tests/test_modeling_tf_common.py
tests/test_modeling_tf_common.py
+3
-2
No files found.
src/transformers/modeling_tf_distilbert.py
View file @
96c49901
...
...
@@ -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__
)
...
...
src/transformers/modeling_tf_openai.py
View file @
96c49901
...
...
@@ -29,7 +29,6 @@ from .modeling_tf_utils import (
TFSequenceSummary
,
TFSharedEmbeddings
,
get_initializer
,
keras_serializable
,
shape_list
,
)
...
...
src/transformers/modeling_tf_roberta.py
View file @
96c49901
...
...
@@ -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__
)
...
...
src/transformers/modeling_tf_t5.py
View file @
96c49901
...
...
@@ -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__
)
...
...
src/transformers/modeling_tf_xlm.py
View file @
96c49901
...
...
@@ -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__
)
...
...
tests/test_modeling_tf_common.py
View file @
96c49901
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment