diff options
Diffstat (limited to 'beliefs/inference/belief_propagation.py')
-rw-r--r-- | beliefs/inference/belief_propagation.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/beliefs/inference/belief_propagation.py b/beliefs/inference/belief_propagation.py index ecd5e9c..37aa437 100644 --- a/beliefs/inference/belief_propagation.py +++ b/beliefs/inference/belief_propagation.py @@ -54,8 +54,8 @@ class BeliefPropagation: # exclude the message sender (either a parent or child) from getting an # outgoing msg from the node to update - parent_ids = node.parents - set([msg_sender_label_id]) - child_ids = node.children - set([msg_sender_label_id]) + parent_ids = set(node.parents) - set([msg_sender_label_id]) + child_ids = set(node.children) - set([msg_sender_label_id]) print("parent_ids:", parent_ids) print("child_ids:", child_ids) |