Support for episodic tasks

[behradrabiei]

Hi Nicklas,

Excellent codebase as always! I was wondering if there are any plans for supporting episodic tasks similar to the original TD-MPC2 codebase?

[nicklashansen]

Hi @behradrabiei, sorry for the late response -- there are no plans to support episodic tasks at this time. It seems pretty feasible to do so but I believe the tasks in MMBench generally would not benefit much from treating them as episodic (except for Atari maybe). Let me know if end up adding such support and I'll happily review it :-)