[BugFix] Batched environments with non empty batch size

[matteobettini]

This PR fixes BUG #773

[codecov]

Codecov Report

Merging #774 (01398d8) into main (578938a) will increase coverage by 0.00%.
The diff coverage is 100.00%.

❗ Current head 01398d8 differs from pull request most recent head 26b974b. Consider uploading reports for the commit 26b974b to get more accurate results

@@           Coverage Diff           @@
##             main     #774   +/-   ##
=======================================
  Coverage   88.73%   88.73%           
=======================================
  Files         123      123           
  Lines       20986    20987    +1     
=======================================
+ Hits        18622    18623    +1     
  Misses       2364     2364           

Flag	Coverage Δ
habitat-gpu	24.97% <0.00%> (-0.01%)	⬇️
linux-brax	29.58% <0.00%> (-0.01%)	⬇️
linux-cpu	85.29% <100.00%> (+0.01%)	⬆️
linux-gpu	86.18% <100.00%> (+<0.01%)	⬆️
linux-jumanji	30.36% <0.00%> (-0.01%)	⬇️
linux-outdeps-gpu	71.91% <100.00%> (+<0.01%)	⬆️
linux-stable-cpu	85.12% <100.00%> (+<0.01%)	⬆️
linux-stable-gpu	85.83% <100.00%> (+0.01%)	⬆️
linux_examples-gpu	42.94% <0.00%> (-0.01%)	⬇️
macos-cpu	84.95% <100.00%> (+<0.01%)	⬆️
olddeps-gpu	75.91% <100.00%> (+<0.01%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
torchrl/envs/vec_env.py	69.42% <100.00%> (+0.05%)	⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[vmoens]

LGTM thanks!

[vmoens]

@matteobettini can I ask you to add a simple test in test_envs.py to test non regression of this feature?

[matteobettini]

I have added the test. This test should be sufficient for this simple modification.

[vmoens]

The lint is failing, let's fix this and merge!

[matteobettini]

Should be fixed