From 0a817d540e954d359879efda2aae8dee303b0865 Mon Sep 17 00:00:00 2001 From: s Date: Wed, 13 Mar 2024 23:10:59 +0100 Subject: [PATCH] added NoneType check --- 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 @@ 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