From fceeeb376d3e92ed034844d341685b4fb6c82e69 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maximilian=20Ke=C3=9Fler?= Date: Sat, 13 May 2023 13:05:58 +0200 Subject: [PATCH] sat: adjust print_model to instances with variable deck sizes --- sat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sat.py b/sat.py index 7e2e699..7a4581a 100644 --- a/sat.py +++ b/sat.py @@ -326,8 +326,8 @@ def print_model(model, cur_game_state, ls: Literals): print('=== move {} ==='.format(m)) print('clues: ' + ''.join(str(i) for i in range(1, 9) if model.get_py_value(ls.clues[m][i]))) print('strikes: ' + ''.join(str(i) for i in range(1, 3) if model.get_py_value(ls.strikes[m][i]))) - print('draw: ' + ', '.join('{}: {}'.format(i, deck[i]) for i in range(cur_game_state.progress, 50) if model.get_py_value(ls.draw[m][i]))) - print('discard: ' + ', '.join('{}: {}'.format(i, deck[i]) for i in range(50) if model.get_py_value(ls.discard[m][i]))) + print('draw: ' + ', '.join('{}: {}'.format(i, deck[i]) for i in range(cur_game_state.progress, cur_game_state.instance.deck_size) if model.get_py_value(ls.draw[m][i]))) + print('discard: ' + ', '.join('{}: {}'.format(i, deck[i]) for i in range(cur_game_state.instance.deck_size) if model.get_py_value(ls.discard[m][i]))) for s in range(0, cur_game_state.instance.num_suits): print('progress {}: '.format(COLOR_INITIALS[s]) + ''.join(str(r) for r in range(1, 6) if model.get_py_value(ls.progress[m][s, r]))) flags = ['discard_any', 'draw_any', 'play', 'play5', 'incr_clues', 'strike', 'extraround', 'dummyturn']