diff --git a/tests/python/contrib/test_ethosu/test_identity_optimizer.py b/tests/python/contrib/test_ethosu/test_identity_optimizer.py index 49bf0258101b3..551860007ec8f 100644 --- a/tests/python/contrib/test_ethosu/test_identity_optimizer.py +++ b/tests/python/contrib/test_ethosu/test_identity_optimizer.py @@ -19,7 +19,10 @@ Test the identity optimizer pass that removes redundant identity operations from the microNPU codegen. """ -import numpy as np +import pytest + +pytest.importorskip("ethosu.vela") + import tensorflow as tf import tvm