diff options
Diffstat (limited to 'tests/test_belief_propagation.py')
-rw-r--r-- | tests/test_belief_propagation.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/test_belief_propagation.py b/tests/test_belief_propagation.py index ef7ffb0..24ee94b 100644 --- a/tests/test_belief_propagation.py +++ b/tests/test_belief_propagation.py @@ -52,7 +52,8 @@ def many_parents_model(many_parents_edges): @pytest.fixture(scope='function') def one_node_model(): - a_node = BernoulliOrNode(label_id='x', children=set(), parents=set()) + a_node = BernoulliOrNode(label_id='x', children=[], parents=[]) + # a_node = BernoulliOrNode(label_id='x', children=set(), parents=set()) return BernoulliOrModel(edges=None, nodes={'x': a_node}) |