Skip to content
Snippets Groups Projects
Commit 083e0f33 authored by Wei Chen's avatar Wei Chen Committed by Tianqi Chen
Browse files

Improve if_parser_enabled (#2372)

parent 426e3bb0
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment