ModuleNotFoundError: No module named 'ai_edge_litert.aot'

[joshua5town]

Following the docs/getting_started.ipynb im continuing getting this error in multipass ->

warnings.warn(
Traceback (most recent call last):
File "/home/ubuntu/android/test_ai_edge.py", line 2, in
import ai_edge_torch
File "/home/ubuntu/env/lib/python3.11/site-packages/ai_edge_torch/init.py", line 16, in
from ai_edge_torch._convert.converter import convert
File "/home/ubuntu/env/lib/python3.11/site-packages/ai_edge_torch/_convert/converter.py", line 21, in
from ai_edge_torch._convert import conversion
File "/home/ubuntu/env/lib/python3.11/site-packages/ai_edge_torch/_convert/conversion.py", line 29, in
from ai_edge_litert.aot import aot_compile as aot_compile_lib
ModuleNotFoundError: No module named 'ai_edge_litert.aot'

Error in Macos m3 pro ->

dlopen(/Users///.venv/lib/python3.11/site-packages/tensorflow/lite/python/metrics/_pywrap_tensorflow_lite_metrics_wrapper.so, 0x0002): Symbol not found: _Wrapped_PyInit__pywrap_tensorflow_lite_metrics_wrapper

any directions to help me get this going?

[gaikwadrahul8]

Hi, @joshua5town
Apologize for the delayed response, I was trying to reproduce the same behavior from my end in Google colab I'm getting below error so we need to dig more into this issue and will update you soon. I see one similar issue reported here :#847 Thank you for bringing this issue to our attention.

---------------------------------------------------------------------------
NotFoundError                             Traceback (most recent call last)
[/tmp/ipython-input-2940186061.py](https://localhost:8080/#) in <cell line: 0>()
      1 import numpy as np
----> 2 import ai_edge_torch
      3 import torch
      4 import torchvision

4 frames
[/usr/local/lib/python3.12/dist-packages/tensorflow/python/framework/load_library.py](https://localhost:8080/#) in load_library(library_location)
    149 
    150     for lib in kernel_libraries:
--> 151       py_tf.TF_LoadLibrary(lib)
    152 
    153   else:

NotFoundError: /usr/local/lib/python3.12/dist-packages/tensorflow/core/kernels/libtfkernel_sobol_op.so: undefined symbol: _ZN10tensorflow15TensorShapeBaseINS_11TensorShapeEEC2EN4absl12lts_202308024SpanIKlEE

[niuchl]

@arfaian could you please take a look at this issue?

[kpeng-bit]

encountered same issue, resolved by making sure docker platform is x86 instead of arm. --platform linux/amd64

[jaewoosong]

Same issue on ARM platform (ModuleNotFoundError: No module named 'ai_edge_litert.aot')