diff --git a/onnxscript/converter.py b/onnxscript/converter.py index dfcddefbd..c6fc406ae 100644 --- a/onnxscript/converter.py +++ b/onnxscript/converter.py @@ -454,7 +454,9 @@ def _eval_constant_expr(self, expr: ast.AST) -> PyValue: # TODO: assert (self._is_constant_expr(expr)) # TODO: Refine types locals: dict[Any, Any] = {} - expr = ast.Expression(expr, lineno=expr.lineno, col_offset=expr.col_offset) + expr = ast.Expression(body=expr) + expr.lineno = expr.body.lineno + expr.col_offset = expr.body.col_offset cpl = compile(expr, filename="", mode="eval") try: return eval(cpl, self.globals, locals) # pylint: disable=eval-used