From 53107cd2371c49d4de4b41b61151f440f1b92a79 Mon Sep 17 00:00:00 2001 From: Alexander Schmidt Date: Tue, 3 Nov 2020 11:41:18 +0100 Subject: [PATCH] refactoring --- src/qtable.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/qtable.py b/src/qtable.py index fb65008..29a3081 100755 --- a/src/qtable.py +++ b/src/qtable.py @@ -40,7 +40,6 @@ def select_option(opts, qs): decision_pt += qs[ opt_qid_pair[1] ] if ran_pt <= decision_pt: return (opt_qid_pair[0], opt_qid_pair[1]) - return (None, None) Q = np.ones([states_dim, actions_dim]) @@ -51,8 +50,9 @@ for i in range(num_episodes): g.dice() state_qid = find_state_qid(g.get_shutable(), g.get_diced()) while not g.is_over(): - opt, opt_qid = select_option( g.get_options(), Q[state_qid, :] ) - if opt: + options = g.get_options() + if len(options) > 0: + opt, opt_qid = select_option( options, Q[state_qid, :] ) old_score = g.get_score() g.shut(opt) g.dice() -- 2.39.2