#2300: scripts: check if comm data exists

DARMA-tasking · Sep 20, 2024 · 4e4eaf4 · 4e4eaf4
1 parent f0563dd
commit 4e4eaf4
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/scripts/JSON_data_files_validator.py b/scripts/JSON_data_files_validator.py
@@ -417,11 +417,10 @@ def validate_comm_links(all_jsons):
             task_ids = set()
 
             for data in all_jsons:
-                #FIXME: KeyErorr: 'communications'
-                #if data... get("communications") is not None:
-                comms = data["phases"][n]["communications"]
-                comm_ids.update({int(comm["from"]["id"]) for comm in comms})
-                comm_ids.update({int(comm["to"]["id"]) for comm in comms})
+                if data["phases"][n].get("communications") is not None:
+                    comms = data["phases"][n]["communications"]
+                    comm_ids.update({int(comm["from"]["id"]) for comm in comms})
+                    comm_ids.update({int(comm["to"]["id"]) for comm in comms})
 
                 tasks = data["phases"][n]["tasks"]
                 task_ids.update({int(task["entity"]["id"]) for task in tasks})