[BUG] BatchedEnv
crashes when EnvBase
has non-empty batch_size
#773
Labels
bug
Something isn't working
BatchedEnv
crashes when EnvBase
has non-empty batch_size
#773
Describe the bug
In the function
get_metadata()
of_BatchedEnv
the tensordictexpand
method is called in a wrong manner:As per docs,
expand
should be called with an argument which is the concatenation of the new dims to the old dimsTo Reproduce
Create a
ParallelEnv
from aBaseEnv
with non-empy batch_sizeThe text was updated successfully, but these errors were encountered: