From ecc90fb9c08bd534b68102352d6a02e240721b7c Mon Sep 17 00:00:00 2001
From: "Matt Welsh (OctoML)" <63477620+mdw-octoml@users.noreply.github.com>
Date: Thu, 28 Jan 2021 21:08:05 -0800
Subject: [PATCH] Some docstring fixes. (#7367)

---
 python/tvm/contrib/graph_runtime.py |  4 ++--
 python/tvm/relay/build_module.py    | 10 ++--------
 2 files changed, 4 insertions(+), 10 deletions(-)

diff --git a/python/tvm/contrib/graph_runtime.py b/python/tvm/contrib/graph_runtime.py
index a960e552f68f..59db716e917c 100644
--- a/python/tvm/contrib/graph_runtime.py
+++ b/python/tvm/contrib/graph_runtime.py
@@ -141,11 +141,11 @@ class GraphModule(object):
         lib = relay.build(...)
         lib.export_library("compiled_lib.so")
         # load it back as a runtime
-        lib:tvm.runtime.Module = tvm.runtime.load_module("compiled_lib.so")
+        lib: tvm.runtime.Module = tvm.runtime.load_module("compiled_lib.so")
         # Call the library factory function for default and create
         # a new runtime.Module, wrap with graph module.
         gmod = graph_runtime.GraphModule(lib["default"](ctx))
-        # use the gmod
+        # use the graph module.
         gmod.set_input("x", data)
         gmod.run()
     """
diff --git a/python/tvm/relay/build_module.py b/python/tvm/relay/build_module.py
index 20cdc24ebc69..f05e105ed2a2 100644
--- a/python/tvm/relay/build_module.py
+++ b/python/tvm/relay/build_module.py
@@ -110,14 +110,8 @@ def build(self, mod, target=None, target_host=None, params=None):
 
         Returns
         -------
-        graph_json : str
-            The json string that can be accepted by graph runtime.
-
-        mod : tvm.Module
-            The module containing necessary libraries.
-
-        params : dict
-            The parameters of the final graph.
+        factory_module : tvm.relay.backend.graph_runtime_factory.GraphRuntimeFactoryModule
+            The runtime factory for the TVM graph runtime.
         """
         target = _update_target(target)