Update.

[picoclvr.git] / escape.py

diff --git a/escape.py b/escape.py
index f2a1662..8066479 100755 (executable)
--- 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)