diff --git a/mlir/dialect.py b/mlir/dialect.py index 3a192ee..fb1cc64 100644 --- a/mlir/dialect.py +++ b/mlir/dialect.py @@ -79,8 +79,8 @@ def make_rules(cls): for fname, ftype in sfields: syntax = syntax.replace('{%s.%s}' % (fname, ftype), ftype) # Replace back braces - syntax = syntax.replace('{LBRACE}', '{') - syntax = syntax.replace('{RBRACE}', '}') + syntax = syntax.replace('{LBRACE}', '"{"') + syntax = syntax.replace('{RBRACE}', '"}"') lark_exprs.append(syntax) cls._fields_ = list(fields)