Merge pull request #23 from zhangyiant/zhangyiant/no-arrays

Handle the case where there's no array field in block template dictio…
zhangyiant · Mar 26, 2022 · a30dfed · a30dfed
2 parents f4c9c16 + c38f021
commit a30dfed
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/register_printer/data_model/block_template.py b/register_printer/data_model/block_template.py
@@ -32,9 +32,10 @@ def generate_block_template(block_template_dict):
             )
             register.add_field(field)
         block_template.add_register(register)
-    for array_dict in block_template_dict["arrays"]:
-        array = ArrayTemplate.from_dict(array_dict)
-        block_template.add_array(array)
+    if "arrays" in block_template_dict:
+        for array_dict in block_template_dict["arrays"]:
+            array = ArrayTemplate.from_dict(array_dict)
+            block_template.add_array(array)
     return block_template