skip tests when required packages are not available

Change-Id: I0a88d92dd31ca3dd07a2a495f18c10a2ebf2fc9e
apache · Feb 16, 2022 · 0db7b03 · 0db7b03
1 parent 4ebb958
commit 0db7b03
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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