diff --git a/optimum/commands/__init__.py b/optimum/commands/__init__.py index 004d1a7263..540ea4dd86 100644 --- a/optimum/commands/__init__.py +++ b/optimum/commands/__init__.py @@ -15,5 +15,5 @@ from .base import BaseOptimumCLICommand, CommandInfo, RootOptimumCLICommand from .env import EnvironmentCommand from .export import ExportCommand, ONNXExportCommand, TFLiteExportCommand -from .onnxruntime import ONNXRuntimeCommand, ONNXRuntimmeOptimizeCommand, ONNXRuntimmeQuantizeCommand +from .onnxruntime import ONNXRuntimeCommand, ONNXRuntimeOptimizeCommand, ONNXRuntimeQuantizeCommand from .optimum_cli import register_optimum_cli_subcommand diff --git a/optimum/commands/onnxruntime/__init__.py b/optimum/commands/onnxruntime/__init__.py index b1112987b4..1b9c24c3b2 100644 --- a/optimum/commands/onnxruntime/__init__.py +++ b/optimum/commands/onnxruntime/__init__.py @@ -14,5 +14,5 @@ # limitations under the License. from .base import ONNXRuntimeCommand -from .optimize import ONNXRuntimmeOptimizeCommand -from .quantize import ONNXRuntimmeQuantizeCommand +from .optimize import ONNXRuntimeOptimizeCommand +from .quantize import ONNXRuntimeQuantizeCommand diff --git a/optimum/commands/onnxruntime/base.py b/optimum/commands/onnxruntime/base.py index 24c64300cc..53e3245ea4 100644 --- a/optimum/commands/onnxruntime/base.py +++ b/optimum/commands/onnxruntime/base.py @@ -15,8 +15,8 @@ """optimum.onnxruntime command-line interface base classes.""" from .. import BaseOptimumCLICommand, CommandInfo -from .optimize import ONNXRuntimmeOptimizeCommand -from .quantize import ONNXRuntimmeQuantizeCommand +from .optimize import ONNXRuntimeOptimizeCommand +from .quantize import ONNXRuntimeQuantizeCommand class ONNXRuntimeCommand(BaseOptimumCLICommand): @@ -28,11 +28,11 @@ class ONNXRuntimeCommand(BaseOptimumCLICommand): CommandInfo( name="optimize", help="Optimize ONNX models.", - subcommand_class=ONNXRuntimmeOptimizeCommand, + subcommand_class=ONNXRuntimeOptimizeCommand, ), CommandInfo( name="quantize", help="Dynammic quantization for ONNX models.", - subcommand_class=ONNXRuntimmeQuantizeCommand, + subcommand_class=ONNXRuntimeQuantizeCommand, ), ) diff --git a/optimum/commands/onnxruntime/optimize.py b/optimum/commands/onnxruntime/optimize.py index 68dac23198..5890e0a07c 100644 --- a/optimum/commands/onnxruntime/optimize.py +++ b/optimum/commands/onnxruntime/optimize.py @@ -69,7 +69,7 @@ def parse_args_onnxruntime_optimize(parser: "ArgumentParser"): ) -class ONNXRuntimmeOptimizeCommand(BaseOptimumCLICommand): +class ONNXRuntimeOptimizeCommand(BaseOptimumCLICommand): @staticmethod def parse_args(parser: "ArgumentParser"): return parse_args_onnxruntime_optimize(parser) diff --git a/optimum/commands/onnxruntime/quantize.py b/optimum/commands/onnxruntime/quantize.py index b82059d4b2..0ce7e6c3dc 100644 --- a/optimum/commands/onnxruntime/quantize.py +++ b/optimum/commands/onnxruntime/quantize.py @@ -63,7 +63,7 @@ def parse_args_onnxruntime_quantize(parser: "ArgumentParser"): ) -class ONNXRuntimmeQuantizeCommand(BaseOptimumCLICommand): +class ONNXRuntimeQuantizeCommand(BaseOptimumCLICommand): @staticmethod def parse_args(parser: "ArgumentParser"): return parse_args_onnxruntime_quantize(parser)