From 620d2e40948d4d3109891bc3f800e91fddc1c033 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Philipp=20Schr=C3=B6er?= Date: Thu, 9 Apr 2020 09:58:47 +0200 Subject: [PATCH] StateGenerator: update test --- tests/storage/test_state_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/storage/test_state_generation.py b/tests/storage/test_state_generation.py index 153a4fd..4c909de 100644 --- a/tests/storage/test_state_generation.py +++ b/tests/storage/test_state_generation.py @@ -30,7 +30,7 @@ def _dfs_explore(program, callback): successors = generator.expand() assert len(successors) <= 1 for choice in successors: - for state_id, _prob in choice: + for state_id, _prob in choice.distribution: queue.push(state_id) current_state_id = queue.pop() if current_state_id is None: