ModuleNotFoundError: No module named 'aimet_common.libpymo' in release1.28.0

[Keywennn]

when I use bash to check the import, there is no error. but when I import directly, I can't find aimet_common.libpymo.

%%bash
sh /usr/local/lib/python3.8/site-packages/aimet_common/bin/envsetup.sh
python -c "
from aimet_common.defs import QuantScheme
import aimet_common.defs as aimet_common_defs
import aimet_common.libpymo as libpymo
"

no error

---

import copy
import torch
from torchvision import models
from aimet_torch.cross_layer_equalization import equalize_model
from aimet_torch import visualize_model

ModuleNotFoundError Traceback (most recent call last)
in <cell line: 4>()
2 import torch
3 from torchvision import models
----> 4 from aimet_torch.cross_layer_equalization import equalize_model
5 from aimet_torch import visualize_model

/usr/local/lib/python3.8/site-packages/aimet_torch/cross_layer_equalization.py in
49 import torch
50
---> 51 import aimet_common.libpymo as libpymo # pylint: disable=import-error
52
53 from aimet_common.utils import AimetLogger

ModuleNotFoundError: No module named 'aimet_common.libpymo'

[quic-mangal]

Can you check if your PYTHONPATH is set correctly?

[canyagmur]

Resolved the issue @480.