Browse Source

StateGenerator: update test

refactoring
Philipp Schröer 5 years ago
parent
commit
620d2e4094
  1. 2
      tests/storage/test_state_generation.py

2
tests/storage/test_state_generation.py

@ -30,7 +30,7 @@ def _dfs_explore(program, callback):
successors = generator.expand() successors = generator.expand()
assert len(successors) <= 1 assert len(successors) <= 1
for choice in successors: for choice in successors:
for state_id, _prob in choice:
for state_id, _prob in choice.distribution:
queue.push(state_id) queue.push(state_id)
current_state_id = queue.pop() current_state_id = queue.pop()
if current_state_id is None: if current_state_id is None:

Loading…
Cancel
Save