Style.

mfuntowicz · mfuntowicz · commit e60c2bb60a2c · 2021-06-29T11:58:33.000+02:00
diff --git a/src/transformers/configuration_utils.py b/src/transformers/configuration_utils.py
@@ -722,4 +722,3 @@ def update_from_string(self, update_str: str):
                 )
 
             setattr(self, k, v)
-
diff --git a/src/transformers/models/albert/configuration_albert.py b/src/transformers/models/albert/configuration_albert.py
@@ -168,4 +168,3 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
         return {"last_hidden_state": {0: "batch", 1: "sequence"}, "pooler_output": {0: "batch"}}
-
diff --git a/src/transformers/models/bart/configuration_bart.py b/src/transformers/models/bart/configuration_bart.py
@@ -210,4 +210,4 @@ def outputs(self) -> Mapping[str, Mapping[int, str]]:
             return {
                 "last_hidden_state": {0: "batch", 1: "sequence"},
                 "encoder_last_hidden_state": {0: "batch", 1: "sequence"},
-            }
+            }
diff --git a/src/transformers/models/bert/configuration_bert.py b/src/transformers/models/bert/configuration_bert.py
@@ -170,4 +170,3 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
         return {"last_hidden_state": {0: "batch", 1: "sequence"}, "pooler_output": {0: "batch"}}
-
diff --git a/src/transformers/models/distilbert/configuration_distilbert.py b/src/transformers/models/distilbert/configuration_distilbert.py
@@ -150,4 +150,3 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
         return {"last_hidden_state": {0: "batch", 1: "sequence"}}
-
diff --git a/src/transformers/models/gpt2/configuration_gpt2.py b/src/transformers/models/gpt2/configuration_gpt2.py
@@ -218,4 +218,4 @@ def outputs(self) -> Mapping[str, Mapping[int, str]]:
             return {
                 "last_hidden_state": {0: "batch", 1: "sequence"},
                 "encoder_last_hidden_state": {0: "batch", 1: "sequence"},
-            }
+            }
diff --git a/src/transformers/models/longformer/__init__.py b/src/transformers/models/longformer/__init__.py
@@ -25,7 +25,7 @@
     "configuration_longformer": [
         "LONGFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP",
         "LongformerConfig",
-        "LongformerOnnxConfig"
+        "LongformerOnnxConfig",
     ],
     "tokenization_longformer": ["LongformerTokenizer"],
 }
@@ -61,11 +61,10 @@
 
 
 if TYPE_CHECKING:
-    from .configuration_longformer import LONGFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, LongformerConfig
     from .configuration_longformer import (
         LONGFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP,
         LongformerConfig,
-        LongformerOnnxConfig
+        LongformerOnnxConfig,
     )
     from .tokenization_longformer import LongformerTokenizer
 
diff --git a/src/transformers/models/longformer/configuration_longformer.py b/src/transformers/models/longformer/configuration_longformer.py
@@ -73,7 +73,6 @@ def __init__(self, attention_window: Union[List[int], int] = 512, sep_token_id:
 
 
 class LongformerOnnxConfig(OnnxConfig):
-
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
         return {
@@ -83,9 +82,4 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
 
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
-        return {
-            "last_hidden_state": {0: "batch", 1: "sequence"},
-            "pooler_output": {0: "batch"}
-        }
-
-
+        return {"last_hidden_state": {0: "batch", 1: "sequence"}, "pooler_output": {0: "batch"}}
diff --git a/src/transformers/models/roberta/configuration_roberta.py b/src/transformers/models/roberta/configuration_roberta.py
@@ -77,4 +77,3 @@ def inputs(self) -> Mapping[str, Mapping[int, str]]:
     @property
     def outputs(self) -> Mapping[str, Mapping[int, str]]:
         return {"last_hidden_state": {0: "batch", 1: "sequence"}, "pooler_output": {0: "batch"}}
-
diff --git a/tests/test_onnx_v2.py b/tests/test_onnx_v2.py
@@ -3,11 +3,17 @@
 from unittest import TestCase
 from unittest.mock import patch
 
-from transformers import AutoTokenizer, is_torch_available, AlbertConfig, DistilBertConfig, LongformerConfig, \
-    RobertaConfig, XLMRobertaConfig
+from transformers import (
+    AlbertConfig,
+    AutoTokenizer,
+    DistilBertConfig,
+    LongformerConfig,
+    RobertaConfig,
+    XLMRobertaConfig,
+    is_torch_available,
+)
 from transformers.models.albert import AlbertOnnxConfig
-
-from transformers.models.bert.configuration_bert import BertOnnxConfig, BertConfig
+from transformers.models.bert.configuration_bert import BertConfig, BertOnnxConfig
 from transformers.models.distilbert import DistilBertOnnxConfig
 from transformers.models.longformer import LongformerOnnxConfig
 from transformers.models.roberta import RobertaOnnxConfig
@@ -21,7 +27,7 @@
     compute_serialized_parameters_size,
     flatten_output_collection_property,
 )
-from transformers.testing_utils import require_onnx, slow, require_torch
+from transformers.testing_utils import require_onnx, require_torch, slow
 
 
 @require_onnx
@@ -81,6 +87,7 @@ def test_use_external_data_format(self):
 
 if is_torch_available():
     from transformers import AlbertModel, BertModel, DistilBertModel, LongformerModel, RobertaModel, XLMRobertaModel
+
     PYTORCH_EXPORT_DEFAULT_MODELS = {
         ("ALBERT", "albert-base-v2", AlbertModel, AlbertConfig, AlbertOnnxConfig),
         ("BERT", "bert-base-cased", BertModel, BertConfig, BertOnnxConfig),

Original file line number	Diff line number	Diff line change
`@@ -722,4 +722,3 @@ def update_from_string(self, update_str: str):`
`722`	`722`	`)`
`723`	`723`
`724`	`724`	`setattr(self, k, v)`
`725`		`-`