Update decorator code to fully use new MiniPython based frontend. (ap…

…ache#20)

* Remove old decorator and port to new one.

* Repair tests

* Add module docstring

* Trigger another build

* Test to see if only this test

relay/python/relay/frontend/__init__.py

@@ -1,3 +1,3 @@
 #pylint: disable-all
 "Frontends for Relay."
-from relay.frontend.relay_decorator import *
+from .mini_python import relay, get_env

relay/python/relay/frontend/mini_python/__init__.py

@@ -0,0 +1,3 @@
+"""The Python frontend for Relay."""
+from .py_to_mini_py import compile_func, get_env
+from .relay_decorator import relay

relay/python/relay/frontend/mini_python/mini_py_to_relay.py

@@ -5,7 +5,7 @@
 import relay.ir as ir
 import relay.make as relay_mk
 import relay.visitor as relay_visit
-import relay.frontend.mini_python.ir as mp
+from . import ir as mp
 
 class MiniPythonToRelay(ast.NodeVisitor):
     bindings: List[Dict[ir.LocalId, Tuple[ir.Type, ir.Expr]]]

relay/python/relay/frontend/mini_python/py_to_mini_py.py

@@ -8,7 +8,7 @@
 import relay.make as relay_mk
 # pylint: disable=unused-import
 from relay.operators import initialize_operators, __relay_tvm_context__
-import relay.frontend.mini_python.ir as mp
+from . import ir as mp
 
 # Contains a global environment of all items.
 __relay_environment__ = relay_mk.Environment({})

relay/python/relay/frontend/mini_python/decorator.py → relay/python/relay/frontend/mini_python/relay_decorator.py

@@ -13,7 +13,7 @@
 from relay.make import *
 from relay.ir import BOp, UOp, Value
 from relay.operators import __relay_tvm_context__
-from .py_to_mini_py import compile_func
+from .py_to_mini_py import compile_func, get_env
 from .mini_py_to_relay import compile_global_defn
 
 def relay_compile(f):
@@ -39,7 +39,6 @@ def marshal_argument(arg, _) -> ir.Value:
     else:
         raise Exception(f"unsupported argument type {type(arg)}")
 
-# TODO(@weberlo): Replace current decorator usage with this one.
 def relay(func):
     """The Relay decorator.
 
@@ -49,7 +48,7 @@ def relay(func):
     """
     env = get_env()
     try:
-        defn = compile_func(func)
+        defn = relay_compile(func)
         env.add(defn)
     except FrontendError as e:
         get_env().display_errors()