Update.

[picoclvr.git] / escape.py

diff --git a/escape.py b/escape.py
index fc4fbbc..f945172 100755 (executable)
--- a/escape.py
+++ b/escape.py
@@ -244,8 +244,8 @@ def episodes2str(
 ######################################################################
 
 if __name__ == "__main__":
-    nb, height, width, T = 8, 4, 6, 20
+    nb, height, width, T = 10, 4, 6, 20
     states, actions, rewards = generate_episodes(nb, height, width, T)
-    seq = episodes2seq(states, actions, rewards, lookahead_delta=5)
+    seq = episodes2seq(states, actions, rewards, lookahead_delta=T)
     s, a, r, lr = seq2episodes(seq, height, width, lookahead=True)
     print(episodes2str(s, a, r, lookahead_rewards=lr, unicode=True, ansi_colors=True))