From 0db7b033246a4111b46cb4dd430d99343bb802f7 Mon Sep 17 00:00:00 2001
From: Luke Hutton <luke.hutton@arm.com>
Date: Tue, 15 Feb 2022 21:51:24 +0000
Subject: [PATCH] skip tests when required packages are not available

Change-Id: I0a88d92dd31ca3dd07a2a495f18c10a2ebf2fc9e
---
 tests/python/contrib/test_ethosu/test_identity_optimizer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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