rename

Author: echarlaix

optimum/exporters/base.py

@@ -69,12 +69,7 @@
 """
 
 
-# TODO: Remove
 class ExportConfig(ABC):
-    pass
-
-
-class ExportersConfig(ABC):
     """
     Base class describing metadata on how to export the model through the ONNX format.
 

optimum/exporters/onnx/base.py

@@ -47,7 +47,7 @@
     is_onnxruntime_available,
     is_transformers_version,
 )
-from ..base import ExportersConfig
+from ..base import ExportConfig
 from .constants import ONNX_DECODER_MERGED_NAME, ONNX_DECODER_NAME, ONNX_DECODER_WITH_PAST_NAME
 from .model_patcher import DecoderModelPatcher, Seq2SeqModelPatcher
 
@@ -100,11 +100,11 @@
 """
 
 
-class OnnxConfig(ExportersConfig):
+class OnnxConfig(ExportConfig):
     DEFAULT_ONNX_OPSET = 11
     VARIANTS = {"default": "The default ONNX variant."}
     DEFAULT_VARIANT = "default"
-    # TODO: move PATCHING_SPECS to ExportersConfig
+    # TODO: move PATCHING_SPECS to ExportConfig
     PATCHING_SPECS: Optional[List["PatchingSpec"]] = None
     _TASK_TO_COMMON_OUTPUTS = {
         "audio-classification": OrderedDict({"logits": {0: "batch_size"}}),

optimum/exporters/tflite/base.py

@@ -27,7 +27,7 @@
 if is_tf_available():
     import tensorflow as tf
 
-from ..base import ExportersConfig
+from ..base import ExportConfig
 
 
 if TYPE_CHECKING:
@@ -115,7 +115,7 @@ def __post_init__(self):
                 self.approach = QuantizationApproach(self.approach)
 
 
-class TFLiteConfig(ExportersConfig, ABC):
+class TFLiteConfig(ExportConfig, ABC):
     """
     Base class for TFLite exportable model describing metadata on how to export the model through the TFLite format.