[tvm] branch main updated: [Hexagon] Skip test if "onnx" module not available (#13585)

[le...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

leandron pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new fbb31fefda [Hexagon] Skip test if "onnx" module not available (#13585)
fbb31fefda is described below

commit fbb31fefdaa3014769256b0260524626b10317b0
Author: Krzysztof Parzyszek <kp...@quicinc.com>
AuthorDate: Fri Dec 9 03:22:25 2022 -0600

    [Hexagon] Skip test if "onnx" module not available (#13585)
---
 tests/python/contrib/test_hexagon/test_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/contrib/test_hexagon/test_models.py b/tests/python/contrib/test_hexagon/test_models.py
index f91b660766..007a704954 100644
--- a/tests/python/contrib/test_hexagon/test_models.py
+++ b/tests/python/contrib/test_hexagon/test_models.py
@@ -28,7 +28,7 @@ from tvm.relay.backend import Executor, Runtime
 
 def get_mobilenet():
     """Download and import mobilenet model with ONNX"""
-    import onnx  # pylint: disable=import-outside-toplevel
+    onnx = pytest.importorskip("onnx")
 
     model_url = "https://github.com/onnx/models/raw/main/vision/classification/mobilenet/model/mobilenetv2-7.onnx"  # pylint: disable=line-too-long
     model_path = tvm.contrib.download.download_testdata(