diff --git a/tests/python/contrib/test_ethosu/test_codegen.py b/tests/python/contrib/test_ethosu/test_codegen.py index 54c235ea38ecc..e88d6a37c291f 100644 --- a/tests/python/contrib/test_ethosu/test_codegen.py +++ b/tests/python/contrib/test_ethosu/test_codegen.py @@ -15,10 +15,12 @@ # specific language governing permissions and limitations # under the License. # pylint: disable=invalid-name, unused-argument +import pytest + +pytest.importorskip("ethosu.vela") import os import numpy as np import pathlib -import pytest import tvm import tvm.micro as micro diff --git a/tests/python/contrib/test_ethosu/test_networks.py b/tests/python/contrib/test_ethosu/test_networks.py index aaad5a3a493b4..76b8512d558d3 100644 --- a/tests/python/contrib/test_ethosu/test_networks.py +++ b/tests/python/contrib/test_ethosu/test_networks.py @@ -15,6 +15,9 @@ # specific language governing permissions and limitations # under the License. # pylint: disable=invalid-name, unused-argument +import pytest + +pytest.importorskip("ethosu.vela") from tests.python.relay.aot.aot_test_utils import ( convert_to_relay, generate_ref_data,