tenstorrent / tt-buda-demos

Repository of model demos using TT-Buda
Other
54 stars 14 forks source link

Adjust import path in model script #76

Closed kamalrajkannan78 closed 4 months ago

kamalrajkannan78 commented 4 months ago

Adjusting import path in model script as old one causes below error during generating yaml configurations

_______________ ERROR collecting tests/test_pytorch_retinanet.py _______________
ImportError while importing test module '/proj_sw/user_dev/kkannan/may23_retinanet_CI_config/pybuda/third_party/buda-model-demos/model_demos/tests/test_pytorch_retinanet.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/usr/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
third_party/buda-model-demos/model_demos/tests/test_pytorch_retinanet.py:2: in <module>
    from cv_demos.retinanet.pytorch_retinanet import run_retinanet_pytorch
third_party/buda-model-demos/model_demos/cv_demos/retinanet/pytorch_retinanet.py:8: in <module>
    from cv_demos.retinanet.model.model_implementation import Model
third_party/buda-model-demos/model_demos/cv_demos/retinanet/model/model_implementation.py:41: in <module>
    import backbone as backbones_mod
E   ModuleNotFoundError: No module named 'backbone'