Update.

[picoclvr.git] / evasion.py

diff --git a/evasion.py b/evasion.py
index 5d9547f..9f37494 100755 (executable)
--- a/evasion.py
+++ b/evasion.py
@@ -184,8 +184,8 @@ def episodes2str(states, actions, rewards, unicode=False, ansi_colors=False):
 ######################################################################
 
 if __name__ == "__main__":
-    height, width, T = 4, 6, 20
-    states, actions, rewards = generate_episodes(3, height, width, T)
+    nb, height, width, T = 8, 4, 6, 20
+    states, actions, rewards = generate_episodes(nb, height, width, T)
     seq = episodes2seq(states, actions, rewards)
     s, a, r = seq2episodes(seq, height, width)
     print(episodes2str(s, a, r, unicode=True, ansi_colors=True))