black format

apache · Mar 19, 2021 · bc9bab8 · bc9bab8
1 parent 5f1e708
commit bc9bab8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/python/tvm/contrib/debugger/debug_runtime.py b/python/tvm/contrib/debugger/debug_runtime.py
@@ -22,6 +22,8 @@
 
 
 def create(*args, **kwargs):
-    warnings.warn("This function has been moved to tvm.contrib.graph_executor and will be removed "
-                  "in the next TVM release")
+    warnings.warn(
+        "This function has been moved to tvm.contrib.graph_executor and will be removed "
+        "in the next TVM release"
+    )
     return debug_executor.create(*args, **kwargs)
diff --git a/python/tvm/contrib/graph_runtime.py b/python/tvm/contrib/graph_runtime.py
@@ -22,6 +22,8 @@
 
 
 def create(*args, **kwargs):
-    warnings.warn("This function has been moved to tvm.contrib.graph_executor and will be removed "
-                  "in the next TVM release")
+    warnings.warn(
+        "This function has been moved to tvm.contrib.graph_executor and will be removed "
+        "in the next TVM release"
+    )
     return graph_executor.create(*args, **kwargs)
diff --git a/python/tvm/relay/build_module.py b/python/tvm/relay/build_module.py
@@ -392,7 +392,9 @@ def _make_executor(self, expr=None):
         self.mod = InferType()(self.mod)
         ret_type = self.mod["main"].checked_type.ret_type
         if _ty.is_dynamic(ret_type):
-            raise ValueError("Graph Executor only supports static graphs, got output type", ret_type)
+            raise ValueError(
+                "Graph Executor only supports static graphs, got output type", ret_type
+            )
         mod = build(self.mod, target=self.target)
         gmodule = _graph_rt.GraphModule(mod["default"](self.ctx))