simplify code

OMR-Research · Jun 11, 2024 · 26b920a · 26b920a
1 parent 33262ba
commit 26b920a
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/mung/io.py b/mung/io.py
@@ -250,18 +250,10 @@ def read_nodes_from_file(filename: str) -> List[Node]:
     root = tree.getroot()
     logging.debug('XML parsed.')
     nodes = []
-    dataset = 'Unknown'
 
-    if 'dataset' in root.attrib:
-        dataset = root.attrib['dataset']
-    document = 'Unknown'
-
-    if 'document' in root.attrib:
-        document = root.attrib['document']
-    node_tag = 'Node'
-
-    if not list(root.iter(node_tag)):
-        node_tag = 'CropObject'
+    dataset = root.attrib.get('dataset', 'Unknown')
+    document = root.attrib.get('document', 'Unknown')
+    node_tag = 'Node' if list(root.iter('Node')) else 'CropObject'
 
     for i, node in enumerate(root.iter(node_tag)):
         ######################################################