- s, a, r = escape.seq2episodes(result, self.height, self.width)
- str = escape.episodes2str(s, a, r, unicode=True, ansi_colors=True)
+ s, a, r, lr = escape.seq2episodes(
+ result, self.height, self.width, lookahead=True
+ )
+ str = escape.episodes2str(
+ s, a, r, lookahead_rewards=lr, unicode=True, ansi_colors=True
+ )