[BUG] `InitTracker` key flexibility about rl HOT 2 CLOSED

For 1)
IIRC yes there was, because we assume that the done is at the root and that the root batch-size is indicative of the env structure.
What would it mean to have a nested "init"?
Happy to see a refactoring if it is well defined.

For 2) I don't know, if the value isn't used anywhere we can remove it

from rl.

The transform seems to be ok with multiple reset and init keys (the only thing it needs to do is just match an init key for any reset key)

It is just the init method that needs some refreshing I think.

from rl.