[BUG] Latest version of TorchRL and TensorDict aren't compatible

[MorganCThomas]

Error

RuntimeError: tensor shape is incompatible with dest shape, got: tensor.shape=torch.Size([128]), dest=torch.Size([3, 512])

Reproduce

acegen-open/tests/check_scripts/run_reinforce_denovo.sh

Environment

torchrl==0.6.0 (5244a90)
tensordict==0.6.0 (8c65dcb)

Discussion

Seems this is a problem with the TransformedEnv adding LSTM/GRU primers, previously the batch size was automatically added as the leading dimension which doesn't seem to happen with this update.

[albertbou92]

I am able to reproduce the error