improve antlr import error message (apache#4888)

neo-ai · Mar 2, 2020 · 91ad6ca · 91ad6ca
1 parent 64ef2e6
commit 91ad6ca
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/python/tvm/relay/_parser.py b/python/tvm/relay/_parser.py
@@ -47,13 +47,6 @@ def __new__(cls, *args, **kwds):
 from . import op
 
 PYTHON_VERSION = sys.version_info.major
-try:
-    from .grammar.py3.RelayVisitor import RelayVisitor
-    from .grammar.py3.RelayParser import RelayParser
-    from .grammar.py3.RelayLexer import RelayLexer
-except ImportError:
-    raise Exception("Couldn't find ANTLR parser. Try building with USE_ANTLR=ON.")
-
 try:
     from antlr4 import InputStream, CommonTokenStream
     from antlr4.error.ErrorListener import ErrorListener
@@ -62,6 +55,14 @@ def __new__(cls, *args, **kwds):
                     "Try running `pip{version} install antlr4-python{version}-runtime`."
                     .format(version=PYTHON_VERSION))
 
+try:
+    from .grammar.py3.RelayVisitor import RelayVisitor
+    from .grammar.py3.RelayParser import RelayParser
+    from .grammar.py3.RelayLexer import RelayLexer
+except ImportError:
+    raise Exception("Couldn't find ANTLR parser. Try building with USE_ANTLR=ON.")
+
+
 sys.setrecursionlimit(10000)
 
 class ParseError(Exception):