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

5
tests/test_modeling_tf_common.py
View File

@@ -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)