From fd6b7ac320006cb4bb57c4c6beced248b18fe970 Mon Sep 17 00:00:00 2001 From: Krzysztof Parzyszek Date: Fri, 4 Aug 2023 06:31:17 -0700 Subject: [PATCH] Fix lint #1 --- python/tvm/script/parser/core/doc.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/script/parser/core/doc.py b/python/tvm/script/parser/core/doc.py index 7d1ff67035a3..1c5241dc8d90 100644 --- a/python/tvm/script/parser/core/doc.py +++ b/python/tvm/script/parser/core/doc.py @@ -414,6 +414,7 @@ def subscript_from_doc(x: doc.Subscript) -> ast.Subscript: ctx=from_doc(x.ctx), ) elif isinstance(x.slice, doc.Tuple): + def remap_dim(doc_item: doc.Expr) -> ast.Expr: ast_item = from_doc(doc_item) if isinstance(ast_item, (ast.Index, ast.Slice)):