X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=evasion.py;h=9f37494a41cbe21722ed744c95aa6d8850e0177f;hb=734db1d1d2a34ccc5258babd483ddb96e22e676e;hp=5d9547f3458f6b55ca9948dfceff056052f14057;hpb=545b06257dd69285c6321d7bd713d819da74953b;p=picoclvr.git

diff --git a/evasion.py b/evasion.py
index 5d9547f..9f37494 100755
--- 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))