Other envs

[wenzhoulyu]

Hi, i'm sorry to be a bother. I wonder how i run my own env in your code. Beacuse i found that there is if (domain, task) in suite.ALL_TASKS:.

[nicklashansen]

Our repo uses DMControl which follow the standard gym api, so it should be fairly straightforward to extend to other environments. You will need to modify the make_env function at

tdmpc/src/env.py

Line 246 in 96cb703

def make_env(cfg):

such that it uses your own environment + wrappers. You might also need to normalize the action space to [-1, 1] if it is not already normalized.

[wenzhoulyu]

thank you so much!