X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=escape.py;h=8066479e3aafd104001773df583dcf2082f41804;hb=1eeba5d817d6e440a93895d42f6e580e9ba273fd;hp=f2a16622d5d6ba7e2396a5e12f87bae88649359c;hpb=2bb362ef385f477da4af7d8679cc94d42cf6c146;p=picoclvr.git

diff --git a/escape.py b/escape.py
index f2a1662..8066479 100755
--- a/escape.py
+++ b/escape.py
@@ -266,11 +266,11 @@ def episodes2str(
 ######################################################################
 
 if __name__ == "__main__":
-    nb, height, width, T, nb_walls = 5, 5, 7, 20, 5
+    nb, height, width, T, nb_walls = 5, 5, 7, 4, 5
     states, actions, rewards = generate_episodes(nb, height, width, T, nb_walls)
     seq = episodes2seq(states, actions, rewards)
     lr, s, a, r = seq2episodes(seq, height, width)
     print(episodes2str(lr, s, a, r, unicode=True, ansi_colors=True))
-    # print()
-    # for s in seq2str(seq):
-    # print(s)
+    print()
+    for s in seq2str(seq):
+        print(s)