Improve error handling for task watchdog list

lasp · Aug 29, 2024 · ab260fc · ab260fc
1 parent 0a48c43
commit ab260fc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/components/task_watchdog/gen/models/task_watchdog_list.py b/src/components/task_watchdog/gen/models/task_watchdog_list.py
@@ -190,8 +190,16 @@ def _resolve_pet_connectors(self):
             )
         # Resolve all pet connector component ids:
         for pet_info in self.watchdog_list.values():
-            pet_info.connector_id = connected_components[pet_info.connector_name]
-            self.watchdog_list[pet_info.connector_name] = pet_info
+            try:
+                pet_info.connector_id = connected_components[pet_info.connector_name]
+                self.watchdog_list[pet_info.connector_name] = pet_info
+            except KeyError:
+                raise ModelException(
+                    "Task_Watchdog Lists specifies connector name: "
+                    + str(pet_info.connector_name)
+                    + " which does not exist in the assembly."
+                )
+
 
         # Sort the watchdog list by index:
         self.watchdog_list = OrderedDict(