Forced elems to be list

HazyResearch · Dec 22, 2018 · e630f34 · e630f34
1 parent ed1e97e
commit e630f34
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pdftotree/utils/pdf/node.py b/pdftotree/utils/pdf/node.py
@@ -36,6 +36,8 @@ def __init__(self, elems):
         """
         Constructor
         """
+        if not isinstance(elems, list):
+            elems = [elems]
         self.elems = elems
         self.sum_elem_bbox = 0
         for elem in elems: