From 083e0f336161aa3fc4c39a8d644d77f0a2a1c552 Mon Sep 17 00:00:00 2001
From: Wei Chen <ipondering.weic@gmail.com>
Date: Mon, 7 Jan 2019 11:38:05 -0800
Subject: [PATCH] Improve if_parser_enabled (#2372)
---
tests/python/relay/test_ir_parser.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/python/relay/test_ir_parser.py b/tests/python/relay/test_ir_parser.py
index c2c83df7e..8789499e5 100644
--- a/tests/python/relay/test_ir_parser.py
+++ b/tests/python/relay/test_ir_parser.py
@@ -2,6 +2,7 @@ import tvm
from tvm import relay
from tvm.relay.parser import enabled
from tvm.relay.ir_pass import alpha_equal
+from nose import SkipTest
from nose.tools import nottest, raises
from numpy import isclose
from typing import Union
@@ -67,7 +68,7 @@ def if_parser_enabled(func):
@wraps(func)
def wrapper():
if not enabled():
- return
+ raise SkipTest("ANTLR is not installed!")
func()
return wrapper
--
GitLab