Fix lint error

apache · May 12, 2022 · 5ce8acb · 5ce8acb
1 parent 52b83f6
commit 5ce8acb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py
@@ -2116,7 +2116,7 @@ def after_unannotate(a: T.handle, b: T.handle) -> None:
 
     def _normalize_block_arg(self, block: Union[BlockRV, str]) -> BlockRV:
         if isinstance(block, str):
-            block = self.get_block(block)
+            return self.get_block(block)
 
         return block