Skip to content

Commit b17cf23

Browse files
authored
Fix torch-mlir-import-onnx entry point (#3965)
Due to the wrong entry point, calling `torch-mlir-import-onnx` currently fails with ``` $ torch-mlir-import-onnx Traceback (most recent call last): File "venv-torch/bin/torch-mlir-import-onnx", line 5, in <module> from torch_mlir.tools.import_onnx import _cli_main ImportError: cannot import name '_cli_main' from 'torch_mlir.tools.import_onnx' (unknown location) ```
1 parent 8fa3bd9 commit b17cf23

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ def build_extension(self, ext):
272272
},
273273
entry_points={
274274
"console_scripts": [
275-
"torch-mlir-import-onnx = torch_mlir.tools.import_onnx:_cli_main",
275+
"torch-mlir-import-onnx = torch_mlir.tools.import_onnx.__main__:_cli_main",
276276
"torch-mlir-opt = torch_mlir.tools.opt.__main__:main",
277277
],
278278
},

0 commit comments

Comments
 (0)