[TIR] Fix lint.

apache · Dec 15, 2020 · 5208d32 · 5208d32
1 parent d43abea
commit 5208d32
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/unittest/test_tir_build.py b/tests/python/unittest/test_tir_build.py
@@ -18,6 +18,7 @@
 from tvm import tir
 from tvm.ir.transform import PassContext
 
+
 def test_scalar_add():
     a = tir.Var("a", "float32")
     b = tir.Var("b", "float32")
@@ -29,10 +30,11 @@ def test_scalar_add():
     pass_ctx = PassContext.current()
     if pass_ctx.config.get("tir.noalias", True):
         func = func.with_attr("tir.noalias", True)
-    mod = tvm.IRModule({'main': func})
+    mod = tvm.IRModule({"main": func})
     func = tvm.build(mod)
     out = func(1.0, 2.0)
     assert out == 3.0
 
+
 if __name__ == "__main__":
     test_scalar_add()