Imitation learning Environment generation

[iamanshulchauhan1]

Hello,

I'm currently working on an imitation learning project and trying to create a custom task. However, it seems like the framework currently only supports task generation via the RL task constructor.

Is support for imitation learning task creation planned in the near future? I'm attempting to build a custom task using the existing example tasks as a reference, but the current structure is quite complex. A more streamlined or modular approach similar to what's available for RL tasks would be extremely helpful for imitation learning as well.

Thanks in advance