Fixes

Author: mandel

src/pdl/pdl_ast.py

@@ -61,7 +61,9 @@ class LocationType(BaseModel):
 class LocalizedExpression(BaseModel):
     """Expression with location information"""
 
-    model_config = ConfigDict(extra="forbid", use_attribute_docstrings=True)
+    model_config = ConfigDict(
+        extra="forbid", use_attribute_docstrings=True, arbitrary_types_allowed=True
+    )
     expr: Any
     location: Optional[LocationType] = None
     pdl_yaml_src: Optional[YamlSource] = None
@@ -106,7 +108,11 @@ class ContributeTarget(StrEnum):
 class Block(BaseModel):
     """Common fields for all PDL blocks."""
 
-    model_config = ConfigDict(extra="forbid", use_attribute_docstrings=True)
+    model_config = ConfigDict(
+        extra="forbid",
+        use_attribute_docstrings=True,
+        arbitrary_types_allowed=True,
+    )
 
     description: Optional[str] = None
     """Documentation associated to the block.

src/pdl/pdl_parser.py

@@ -41,7 +41,7 @@ def parse_str(pdl_str: str, file_name: str = "") -> tuple[Program, LocationType]
 
 
 def set_program_location(prog: Program, pdl_str: str, file_name: str = ""):
-    loc = strictyaml.dirty_load(pdl_str)
+    loc = strictyaml.dirty_load(pdl_str, allow_flow_style=True)
     set_location(prog.root, loc)