added NoneType check (#16)

specklesystems · Mar 14, 2024 · 4b9d6b0 · 4b9d6b0
1 parent 35ca711
commit 4b9d6b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flatten.py b/flatten.py
@@ -7,7 +7,7 @@
 
 def flatten_base(base: Base) -> Iterable[Base]:
     """Take a base and flatten it to an iterable of bases."""
-    if hasattr(base, "elements"):
+    if hasattr(base, "elements") and base.elements is not None:
         for element in base["elements"]:
             yield from flatten_base(element)
     yield base