From 21c765ca420fc27e705ebdfcd55f4d2b64a1e3fc Mon Sep 17 00:00:00 2001 From: Adrian Boguszewski Date: Fri, 12 Apr 2024 19:12:53 +0200 Subject: [PATCH] Rename mo with ov Signed-off-by: Adrian Boguszewski --- src/anomalib/cli/cli.py | 2 +- src/anomalib/cli/utils/openvino.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/anomalib/cli/cli.py b/src/anomalib/cli/cli.py index 9de4a1d9c1..524e20a90e 100644 --- a/src/anomalib/cli/cli.py +++ b/src/anomalib/cli/cli.py @@ -234,7 +234,7 @@ def add_export_arguments(self, parser: ArgumentParser) -> None: added = parser.add_method_arguments( Engine, "export", - skip={"mo_args", "model"}, + skip={"ov_args", "model"}, ) self.subcommand_method_arguments["export"] = added add_openvino_export_arguments(parser) diff --git a/src/anomalib/cli/utils/openvino.py b/src/anomalib/cli/utils/openvino.py index 70e329f6b4..65ac7b80db 100644 --- a/src/anomalib/cli/utils/openvino.py +++ b/src/anomalib/cli/utils/openvino.py @@ -22,11 +22,11 @@ def add_openvino_export_arguments(parser: ArgumentParser) -> None: """Add OpenVINO arguments to parser under --mo key.""" if get_common_cli_parser is not None: group = parser.add_argument_group("OpenVINO Model Optimizer arguments (optional)") - mo_parser = get_common_cli_parser() + ov_parser = get_common_cli_parser() # remove redundant keys from mo keys - for arg in mo_parser._actions: # noqa: SLF001 + for arg in ov_parser._actions: # noqa: SLF001 if arg.dest in ("help", "input_model", "output_dir"): continue - group.add_argument(f"--mo_args.{arg.dest}", type=arg.type, default=arg.default, help=arg.help) + group.add_argument(f"--ov_args.{arg.dest}", type=arg.type, default=arg.default, help=arg.help) else: logger.info("OpenVINO is possibly not installed in the environment. Skipping adding it to parser.")