|
@ -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: |
|
|