aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorCathy Yeh <cathy@driver.xyz>2017-11-20 11:40:02 -0800
committerCathy Yeh <cathy@driver.xyz>2017-11-20 11:40:02 -0800
commit71e384a741e52f94882b14062a3dc10e5f391533 (patch)
tree669b8c78e3c9c7e44cf58692fef81836b8cc94b9 /tests
parentb16e990b7e4d00e427d4445ba38eef0fb967963a (diff)
downloadbeliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.gz
beliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.bz2
beliefs-71e384a741e52f94882b14062a3dc10e5f391533.zip
BernoulliOrCPD inherits from TabularCPD
Diffstat (limited to 'tests')
-rw-r--r--tests/test_belief_propagation.py3
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})