[Feature] Passing lists of keyword arguments in reset for batched envs

[vmoens]

cc @teopir

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/2076

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 8 New Failures

As of commit a1d3b76 with merge base 02c8342 ():

NEW FAILURES - The following jobs have failed:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  Process completed with exit code 4.
• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
  Process completed with exit code 4.
• Habitat Tests on Linux / tests (3.9, 11.6) / linux-job (gh)
  RuntimeError: Command docker exec -t 1b343603e37fe8a8113f84b734c838f9a5093043e68a3d22970ba23f3f849d8b /exec failed with exit code 139
• Libs Tests on Linux / unittests-brax (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 4d7891532fe13aca464f55ee8e4024d5ae42cee0bb375e1c699d613e69a92278 /exec failed with exit code 1
• Libs Tests on Linux / unittests-robohive (3.9, 12.1) / linux-job (gh)
  test/test_libs.py::TestRoboHive::test_robohive[False]
• Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job (gh)
  test/test_rb.py::TestRBMultidim::test_rb_multidim_collector[env_device0-transform1-sampler_cls4-rbtype1-LazyTensorStorage-TensorDictRoundRobinWriter]
• Unit-tests on MacOS CPU / tests (3.11) / macos-job (gh)
  test/test_modules.py::TestMultiAgent::test_multiagent_mlp[batch1-None-False-True-3]
• Unit-tests on Windows / unittests-cpu / windows-job (gh)
  test/test_transforms.py::TestBatchSizeTransform::test_trans_parallel_env_check[False-reshape]

This comment was automatically generated by Dr. CI and updates every 15 minutes.