From 4b9d6b051d016ca84cb5971d63e4122ababcd1b7 Mon Sep 17 00:00:00 2001 From: s Date: Thu, 14 Mar 2024 12:19:41 +0100 Subject: [PATCH] added NoneType check (#16) --- flatten.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flatten.py b/flatten.py index 4c2d5bb..c180564 100644 --- a/flatten.py +++ b/flatten.py @@ -7,7 +7,7 @@ from specklepy.objects import Base 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