Remove unnecessary imports in __init__

LysandreJik · mfuntowicz · commit 345f5b61df09 · 2021-06-28T20:08:09.000+02:00
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -60,7 +60,7 @@
 
 # Base objects, independent of any specific backend
 _import_structure = {
-    "configuration_utils": ["OnnxConfig", "OnnxVariable", "PretrainedConfig"],
+    "configuration_utils": ["PretrainedConfig"],
     "data": [
         "DataProcessor",
         "InputExample",
@@ -1640,7 +1640,7 @@
 # Direct imports for type-checking
 if TYPE_CHECKING:
     # Configuration
-    from .configuration_utils import OnnxConfig, OnnxVariable, PretrainedConfig
+    from .configuration_utils import PretrainedConfig
 
     # Data
     from .data import (
diff --git a/src/transformers/onnx/__init__.py b/src/transformers/onnx/__init__.py
@@ -1,11 +1,3 @@
 # flake8: noqa
 
 from .config import OnnxConfig, OnnxVariable
-from .convert import convert_pytorch, ensure_model_and_config_inputs_match, optimize, validate_model_outputs
-from .interpolate import (
-    evaluate_expr_to_int,
-    expand_repeated_onnx_variables,
-    insert_additional_onnx_value_within_inputs,
-    interpolate_expression,
-)
-from .utils import flatten_output_collection_property, generate_identified_filename
