-
Notifications
You must be signed in to change notification settings - Fork 2.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
5 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -105,7 +105,7 @@ def load_jsonc(basename: str): | |
return json.loads("\n".join(lines)) | ||
|
||
|
||
def create_backend_test(devices:list[str] = None, test_name=None): | ||
def create_backend_test(devices: list[str] = None, test_name=None): | ||
Check warning Code scanning / lintrunner RUFF/RUF013 Warning test
PEP 484 prohibits implicit Optional.
See https://docs.astral.sh/ruff/rules/implicit-optional |
||
"""Creates an OrtBackendTest and adds its TestCase's to global scope so unittest will find them.""" | ||
|
||
overrides = load_jsonc("onnx_backend_test_series_overrides.jsonc") | ||
|
@@ -198,10 +198,10 @@ def parse_args(): | |
parser.add_argument( | ||
"--devices", | ||
type=str, | ||
choices=["CPU", "CUDA", "MIGRAPHX", "DNNL", "DML", "OPENVINO_GPU", "OPENVINO_CPU", "OPENVINO_NPU","OPENVINO"], | ||
choices=["CPU", "CUDA", "MIGRAPHX", "DNNL", "DML", "OPENVINO_GPU", "OPENVINO_CPU", "OPENVINO_NPU", "OPENVINO"], | ||
nargs="+", # allows multiple values | ||
default=["CPU"], # default to ["CPU"] if no input is given | ||
help="Select one or more devices CPU, CUDA, MIGRAPHX, DNNL, DML, OPENVINO_GPU, OPENVINO_CPU, OPENVINO_NPU, OPENVINO" | ||
help="Select one or more devices CPU, CUDA, MIGRAPHX, DNNL, DML, OPENVINO_GPU, OPENVINO_CPU, OPENVINO_NPU, OPENVINO", | ||
) | ||
|
||
# parse just our args. python unittest has its own args and arg parsing, and that runs inside unittest.main() | ||
|
@@ -214,5 +214,5 @@ def parse_args(): | |
if __name__ == "__main__": | ||
args = parse_args() | ||
|
||
create_backend_test(args.devices,args.test_name) | ||
unittest.main() | ||
create_backend_test(args.devices, args.test_name) | ||
unittest.main() |