+
+def save_seq_as_anim_script(seq, filename):
+ it_len = height * width + 3
+
+ seq = (
+ seq.reshape(seq.size(0), -1, it_len)
+ .permute(1, 0, 2)
+ .reshape(T, seq.size(0), -1)
+ )
+
+ with open(filename, "w") as f:
+ for t in range(T):
+ f.write("clear\n")
+ f.write("cat << EOF\n")
+ # for i in range(seq.size(2)):
+ # lr, s, a, r = seq2episodes(seq[t : t + 1, :, i], height, width)
+ lr, s, a, r = seq2episodes(
+ seq[t : t + 1, :].reshape(5, 10 * it_len), height, width
+ )
+ f.write(episodes2str(lr, s, a, r, unicode=True, ansi_colors=True))
+ f.write("EOF\n")
+ f.write("sleep 0.5\n")
+
+