diff options
| author | Cathy Yeh <cathy@driver.xyz> | 2017-11-13 15:34:33 -0800 |
|---|---|---|
| committer | Cathy Yeh <cathy@driver.xyz> | 2017-11-17 13:48:32 -0800 |
| commit | b16e990b7e4d00e427d4445ba38eef0fb967963a (patch) | |
| tree | 5ecb0137fde4779c0158eaf66f89a8df6288d250 /tests | |
| parent | 77d8b323d4f6e05ca97d9cbef43ac85fd8040d61 (diff) | |
| download | beliefs-b16e990b7e4d00e427d4445ba38eef0fb967963a.tar.gz beliefs-b16e990b7e4d00e427d4445ba38eef0fb967963a.tar.bz2 beliefs-b16e990b7e4d00e427d4445ba38eef0fb967963a.zip | |
changes to work with bump from networkx 1.11 to 2.0
some nx functions now return iterators
Diffstat (limited to 'tests')
| -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 223ba78..ef7ffb0 100644 --- a/tests/test_belief_propagation.py +++ b/tests/test_belief_propagation.py @@ -166,7 +166,8 @@ def test_belief_propagation_modify_model_inplace(simple_model): _ = infer.query(evidence={}) assert simple_model.all_nodes_are_fully_initialized - beliefs_from_model = {node_id: node.belief[1] for node_id, node in simple_model.nodes.items()} + beliefs_from_model = {node_id: node.belief[1] for + node_id, node in simple_model.nodes_dict.items()} compare_dictionaries(expected, beliefs_from_model) |