Temporary fix for non existing parent_uid bug

NVIDIA · Oct 18, 2024 · cdefd87 · cdefd87
1 parent c038b08
commit cdefd87
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nemoguardrails/colang/v2_x/runtime/statemachine.py b/nemoguardrails/colang/v2_x/runtime/statemachine.py
@@ -2422,5 +2422,7 @@ def _is_child_activated_flow(state: State, flow_state: FlowState) -> bool:
     return (
         flow_state.activated > 0
         and flow_state.parent_uid is not None
+        and flow_state.parent_uid
+        in state.flow_states  # TODO: Figure out why this can fail sometimes
         and flow_state.flow_id == state.flow_states[flow_state.parent_uid].flow_id
     )