Fix hierarchy generation inference logic

penpot · Sep 25, 2024 · 1939250 · 1939250
1 parent 4610f65
commit 1939250
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/penai/hierarchy_generation/inference.py b/src/penai/hierarchy_generation/inference.py
@@ -172,6 +172,8 @@ def infer_shape_hierarchy(
             vis.label.replace("#", ""): vis.shape for vis in visualizations
         }
 
+        queried_hierarchy = InferencedHierarchySchema(**queried_hierarchy)
+
         hierarchy = HierarchyElement.from_hierarchy_schema(
             label_shape_mapping, queried_hierarchy
         )