[Feature] More flexibility in loading PettingZoo

[matteobettini]

This change is a PettingZoo change which is BC-compatible to all the pettingzoo versions we already supported.

In addition, in case pettingzoo cannot load all modules, it leverages the new PettingZoo 1.24.3 functionalities to load only group of environments

This ameliorates the issue Farama-Foundation/PettingZoo#1072

And fixes facebookresearch/BenchMARL#41

cc @elliottower

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

⏳ 1 Pending, 5 Unrelated Failures

As of commit 037f72a with merge base e98ee38 ():

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  Workflow failed! Resource not accessible by integration
• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
  Workflow failed! Resource not accessible by integration
• Unit-tests on Linux / tests-stable-gpu (3.8, 11.8) / linux-job (gh)
  test/test_libs.py::TestGym::test_vecenvs_env[CartPole-v1]
• Unit-tests on MacOS CPU / tests (3.11) / macos-job (gh)
  test/test_libs.py::TestGym::test_vecenvs_env[CartPole-v1]

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Habitat Tests on Linux / tests (3.9, 11.6) / linux-job (gh)
  test/test_libs.py::TestHabitat::test_habitat_render[False-HabitatPick-v0]

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

[vmoens]

LGTM thanks