Fix lint.

mfuntowicz · mfuntowicz · commit fa083754549b · 2021-06-29T12:01:04.000+02:00
diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py
@@ -317,10 +317,6 @@ def is_coloredlogs_available():
     return _coloredlogs_available
 
 
-def is_sympy_available():
-    return _sympy_available
-
-
 def is_keras2onnx_available():
     return _keras2onnx_available
 
diff --git a/src/transformers/onnx/__main__.py b/src/transformers/onnx/__main__.py
@@ -17,7 +17,6 @@
 from typing import Callable, Tuple, Union
 
 from onnxruntime import GraphOptimizationLevel
-from transformers.file_utils import is_coloredlogs_available, is_sympy_available
 from transformers.models.albert import AlbertOnnxConfig
 from transformers.models.auto import AutoTokenizer
 from transformers.models.bart import BartOnnxConfig
diff --git a/src/transformers/onnx/convert.py b/src/transformers/onnx/convert.py
@@ -19,8 +19,6 @@
 import numpy as np
 from packaging.version import Version, parse
 
-from onnxruntime import GraphOptimizationLevel
-
 from .. import PreTrainedModel, PreTrainedTokenizer, TensorType, TFPreTrainedModel, is_torch_available
 from ..utils import logging
 from .config import OnnxConfig
diff --git a/src/transformers/onnx/utils.py b/src/transformers/onnx/utils.py
@@ -14,7 +14,6 @@
 
 from ctypes import c_float, sizeof
 from enum import Enum
-from pathlib import Path
 from typing import Any, Dict, Iterable
 
 
diff --git a/tests/test_onnx_v2.py b/tests/test_onnx_v2.py
@@ -7,20 +7,19 @@
     AlbertConfig,
     AutoTokenizer,
     DistilBertConfig,
-    LongformerConfig,
+    # LongformerConfig,
     RobertaConfig,
     XLMRobertaConfig,
     is_torch_available,
 )
 from transformers.models.albert import AlbertOnnxConfig
 from transformers.models.bert.configuration_bert import BertConfig, BertOnnxConfig
 from transformers.models.distilbert import DistilBertOnnxConfig
-from transformers.models.longformer import LongformerOnnxConfig
+# from transformers.models.longformer import LongformerOnnxConfig
 from transformers.models.roberta import RobertaOnnxConfig
 from transformers.models.xlm_roberta import XLMRobertaOnnxConfig
 from transformers.onnx import EXTERNAL_DATA_FORMAT_SIZE_LIMIT, OnnxConfig, ParameterFormat
 
-# from transformers.onnx.convert import convert_pytorch
 from transformers.onnx.config import DEFAULT_ONNX_OPSET
 from transformers.onnx.utils import (
     compute_effective_axis_dimension,
