- 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
- )
+ s, a, r, lr = escape.seq2episodes(result, self.height, self.width)
+ str = escape.episodes2str(lr, s, a, r, unicode=True, ansi_colors=True)