diff --git a/python/tvm/script/parser/core/evaluator.py b/python/tvm/script/parser/core/evaluator.py index 3b940fa6d227..04605fd1ecf4 100644 --- a/python/tvm/script/parser/core/evaluator.py +++ b/python/tvm/script/parser/core/evaluator.py @@ -222,6 +222,8 @@ def _visit(self, node: doc.AST) -> Any: doc.Constant | doc.expr_context | doc.operator | doc.boolop | doc.unaryop | doc.cmpop, ): return node + if isinstance(node, doc.keyword): + return doc.keyword(arg=node.arg, value=self._visit(node.value)) if not isinstance(node, doc.expr | doc.Slice): return node if isinstance(node, doc.Lambda):