Skip to content

Commit

Permalink
Fix usability checker CoreML config file path. (#21626)
Browse files Browse the repository at this point in the history
Fix usability checker CoreML config file path. The files got renamed but one place was still referring to the old name.
  • Loading branch information
edgchen1 authored and prathikr committed Aug 9, 2024
1 parent 10032f2 commit 5db42fa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tools/python/util/mobile_helpers/usability_checker.py
Original file line number Diff line number Diff line change
Expand Up @@ -513,11 +513,11 @@ def check_nnapi_partitions(model, require_fixed_input_sizes: bool):
return _check_ep_partitioning(model, config_path, require_fixed_input_sizes)


def check_coreml_partitions(model: onnx.ModelProto, require_fixed_input_sizes: bool, config_filename):
def check_coreml_partitions(model: onnx.ModelProto, require_fixed_input_sizes: bool, config_filename: str):
# if we're running in the ORT python package the file should be local. otherwise assume we're running from the
# ORT repo
script_dir = pathlib.Path(__file__).parent
local_config = script_dir / "coreml_supported_ops.md"
local_config = script_dir / config_filename
if local_config.exists():
config_path = local_config
else:
Expand Down

0 comments on commit 5db42fa

Please sign in to comment.