Skip to content

Commit 710f344

Browse files
committed
NXP Backend: Use absolute import for experimental CifarNet model in aot_neutron_compile.py.
As of the CifarNet model is installed as part of the ExecuTorch, by previous commit, use absolute import instead of relative.
1 parent e131a9f commit 710f344

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

examples/nxp/aot_neutron_compile.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@
2222
from executorch.examples.models import MODEL_NAME_TO_MODEL
2323
from executorch.examples.models.model_factory import EagerModelFactory
2424

25+
from executorch.examples.nxp.experimental.cifar_net.cifar_net import (
26+
CifarNet,
27+
test_cifarnet_model,
28+
)
29+
2530
from executorch.exir import (
2631
EdgeCompileConfig,
2732
ExecutorchBackendConfig,
@@ -32,8 +37,6 @@
3237
from torch.export import export
3338
from torchao.quantization.pt2e.quantize_pt2e import convert_pt2e, prepare_pt2e
3439

35-
from .experimental.cifar_net.cifar_net import CifarNet, test_cifarnet_model
36-
3740
FORMAT = "[%(levelname)s %(asctime)s %(filename)s:%(lineno)s] %(message)s"
3841
logging.basicConfig(level=logging.INFO, format=FORMAT)
3942

0 commit comments

Comments
 (0)