-def generate_episodes(nb, height=6, width=6, T=10):
+def action2code(r):
+ return first_actions_code + r
+
+
+def code2action(r):
+ return r - first_actions_code
+
+
+def reward2code(r):
+ return first_rewards_code + r + 1
+
+
+def code2reward(r):
+ return r - first_rewards_code - 1
+
+
+def lookahead_reward2code(r):
+ return first_lookahead_rewards_code + r + 1
+
+
+def code2lookahead_reward(r):
+ return r - first_lookahead_rewards_code - 1
+
+
+######################################################################
+
+
+def generate_episodes(nb, height=6, width=6, T=10, nb_walls=3):