From 393088247ea689b4385fc27dfcb516e3eb9e0905 Mon Sep 17 00:00:00 2001
From: Kristian Klemon <k.klemon@appliedai.de>
Date: Wed, 18 Sep 2024 08:35:23 +0200
Subject: [PATCH] Add missing return statement

---
 src/penai/hierarchy_generation/inference.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/penai/hierarchy_generation/inference.py b/src/penai/hierarchy_generation/inference.py
index ebdc347..43a4dba 100644
--- a/src/penai/hierarchy_generation/inference.py
+++ b/src/penai/hierarchy_generation/inference.py
@@ -72,6 +72,8 @@ def from_penpot_shape(cls, shape: PenpotShapeElement) -> Self:
         for child in shape.get_direct_children_shapes():
             element.children.append(cls.from_penpot_shape(child))
 
+        return element
+
     def flatten(self) -> Iterable[Self]:
         yield self