aimet-onnx wheel installation issue

[NikilXYZ]

When I install aimet-onnx (via poetry) to my pyenv virtualenv, aimet_onnx and aimet_common should be folders at .pyenv/versions/<virtual env name>/lib/python3.10/site-packages/.

<virtual_env>/lib/python3.10/site-packages/
│
├─ aimet_onnx
├─ aimet_common

However, the aimet_onnx whl file is doing the following

<virtual_env>/lib/python3.10/site-packages/
│
├─ aimet_onnx-1.35.0+cu118.data
         │
         ├─ purelib
                 │
                 ├─ aimet_onnx
                 ├─ aimet_common

As a result, my python interpreter is not able to find the packages unless I add the subdirectory to my python path or copy those folders out, which I shouldn't have to do.

The aimet-torch from pypi doesn't have this problem. Please fix your packaging.

[quic-kyunggeu]

Hi @NikilXYZ thanks for filing this issue.
It looks like you were using aimet-onnx 1.35.
I didn't manage to reproduce this issue (+ we never encountered this issue in regular pip/conda install), but I'd like to recommend retrying with the latest aimet-onnx 2.x availalbe on pypi.
You can just try poetry add aimet-onnx -- I can at least confirm that it worked in my linux/x86_64/cpython3.10 environment.