[Feature] MCTS policy

[vmoens]

Stack from ghstack (oldest at bottom):

• -> [Feature] MCTS policy #2359
• [Feature] MCTS Scoring functions #2358
• [Feature] TensorSpec.enumerate() #2354
• [Feature] MCTSForest #2307
• [Feature] TensorDictMap #2306
• [Feature] TensorDictMap Query module #2305
• [Feature] TensorDictMap hashing functions #2304

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 4 New Failures, 3 Cancelled Jobs, 16 Unrelated Failures

As of commit 58d5e16 with merge base 35a1c5b ():

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 1.
• Habitat Tests on Linux / tests (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t f3ac66e4ed11d687860f2204c424cf71265e20d83b1c9159a8711d97e7c7160e /exec failed with exit code 139
• Lint / python-source-and-configs / linux-job (gh)
  torchrl/modules/mcts/scores.py:14:1: F401 'torch.nn' imported but unused

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• Wheels / build-wheel-windows (3.10, 3.10.3) (gh)
  Process completed with exit code 1.
• Wheels / build-wheel-windows (3.12, 3.12) (gh)
• Wheels / build-wheel-windows (3.9, 3.9) (gh)
  Process completed with exit code 1.

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

• Build Windows Wheels / pytorch/rl (pytorch/rl, python packaging/wheel/relocate.py, test/smoke_test.py, torchrl) / upload / wheel-py3_9-cpu (gh) (similar failure)
  Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl (pytorch/rl, python packaging/wheel/relocate.py, test/smoke_test.py, torchrl) / upload / wheel-py3_9-cuda11_8 (gh) (similar failure)
  Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl (pytorch/rl, python packaging/wheel/relocate.py, test/smoke_test.py, torchrl) / upload / wheel-py3_9-cuda12_4 (gh) (detected as infra flaky with no log or failing log classifier)
• Wheels / build-wheel-windows (3.11, 3.11) (gh) (matched win rule in flaky-rules.json)
  ##[error]The operation was canceled.
• Wheels / build-wheel-windows (3.8, 3.8) (gh) (similar failure)

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

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

• Build Windows Wheels / pytorch/rl (pytorch/rl, python packaging/wheel/relocate.py, test/smoke_test.py, torchrl) / upload / wheel-py3_9-cuda12_1 (gh) (trunk failure)
  Unable to find any artifacts for the associated workflow
• Unit-tests on Linux / tests-cpu (3.10) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-cpu (3.11) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-cpu (3.8) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-cpu (3.9) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-gpu (3.11, 12.1) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-optdeps (3.10, 12.1) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Linux / tests-stable-gpu (3.10, 11.8) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestRenameTransform::test_rename_done_reward[False]
• Unit-tests on Windows / unittests-cpu / windows-job (gh) (trunk failure)
  Process completed with exit code 1.

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