[Feature] TensorSpec.enumerate()

[vmoens]

Stack from ghstack (oldest at bottom):

• [Feature] MCTS Scoring functions #2358
• -> [Feature] TensorSpec.enumerate() #2354
• [Feature] MCTSForest #2307

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 21 New Failures, 6 Unrelated Failures

As of commit 491c9c8 with merge base 4736fac ():

NEW FAILURES - The following jobs have failed:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  E ImportError: cannot import name 'ListStorage' from partially initialized module 'torchrl.data' (most likely due to a circular import) (/home/runner/work/rl/rl/torchrl/data/__init__.py)
• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
  E ImportError: cannot import name 'ListStorage' from partially initialized module 'torchrl.data' (most likely due to a circular import) (/__w/rl/rl/torchrl/data/__init__.py)
• Examples Tests on Linux / tests (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t aec12c410f3fa0e54be4dd454d9adc0a694392be10931c7e09fe485ed92fff2f /exec failed with exit code 1
• Generate documentation / build-docs (3.10, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 62438979de2a28eb60fa2cdcc2a447b1d26f92ef1ddf47953f2909bc0804987c /exec failed with exit code 1
• Habitat Tests on Linux / tests (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t cf4616a2ff05e7fa08e22b4e47ab9bae74a1cd1539235030acb8ced5cea98c65 /exec failed with exit code 1
• Libs Tests on Linux / unittests-gym (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 4fd7a6971e3cf449869d9277c28aea7d7929815731d17507d7fa4e313d257f23 /exec failed with exit code 1
• Libs Tests on Linux / unittests-sklearn (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 230a42d0204040df44fd56b3667d0f32967de99580de71a8d9fa1efc2929b9ac /exec failed with exit code 1
• Lint / python-source-and-configs / linux-job (gh)
  RuntimeError: Command docker exec -t 7a405f7d4c7b3d29d7e1378c31341b772d90844ea332b3537e1a48ab753c8e02 /exec failed with exit code 1
• RLHF Tests on Linux / unittests (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 9bc9063a6ed43983796341d46bb8b775594c034e7f5d27c869faf13f6efadb56 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.10) / linux-job (gh)
  RuntimeError: Command docker exec -t 05267219b2e5fcac0bb353c41a9c4734e2cba4e2cd4a5a428ebbd10bcf2049c2 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.11) / linux-job (gh)
  RuntimeError: Command docker exec -t 7edd0b9b8675f52c6156a35c2b268e632f45f2e88c0e6563534c6cc892ff3be9 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh)
  RuntimeError: Command docker exec -t 9d982660cb6df505317aa85a5140dfc0d64d0c23ec68bf813b4572052493b423 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.9) / linux-job (gh)
  RuntimeError: Command docker exec -t 3bf3eed6c326de284f84c3302f081f11e234c4daed0ba01d60b86b6455057369 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu-oldget (3.12) / linux-job (gh)
  RuntimeError: Command docker exec -t 2620b288ca8ef0a0381de69805b74ab53b60bf750a0325701554877e4b99c97c /exec failed with exit code 1
• Unit-tests on Linux / tests-gpu (3.11, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 82ded98e570ddf95747d10fd678f9321319ec868d9a2f0f5619bc53217b4cfff /exec failed with exit code 1
• Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job (gh)
  RuntimeError: Command docker exec -t fcc66a6b8a01c8e458d38d7b29ff2bca888aaf936a922a844be05a06227d55c6 /exec failed with exit code 1
• Unit-tests on Linux / tests-optdeps (3.11, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t 8b5bfeabf211899fa7af8841f72ee43111a6ba627d1980430ec55f0475e79a2e /exec failed with exit code 1
• Unit-tests on Linux / tests-stable-gpu (3.10, 11.8) / linux-job (gh)
  RuntimeError: Command docker exec -t 5d7cae6e68d59e9665abf008452a58a1a0320dc46b0d6622fb81c2362ef49c28 /exec failed with exit code 1
• Unit-tests on Windows / unittests-cpu / windows-job (gh)
  ImportError: cannot import name 'ListStorage' from partially initialized module 'torchrl.data' (most likely due to a circular import) (C:\actions-runner\_work\rl\rl\pytorch\rl\torchrl\data\__init__.py)
• Wheels / test-wheel-windows (3.11) (gh)
  ImportError: cannot import name 'ListStorage' from partially initialized module 'torchrl.data' (most likely due to a circular import) (C:\hostedtoolcache\windows\Python\3.11.9\x64\Lib\site-packages\torchrl\data\__init__.py)
• Wheels / test-wheel-windows (3.12) (gh)
  ImportError: cannot import name 'ListStorage' from partially initialized module 'torchrl.data' (most likely due to a circular import) (C:\hostedtoolcache\windows\Python\3.12.7\x64\Lib\site-packages\torchrl\data\__init__.py)

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

• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda12_4 (gh) (similar failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Wheels / test-wheel-windows (3.10) (gh) (matched win rule in flaky-rules.json)
  ##[error]The operation was canceled.
• Wheels / test-wheel-windows (3.9) (gh) (matched win rule in flaky-rules.json)
  ##[error]The operation was canceled.

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 / upload / wheel-py3_9-cpu (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda11_8 (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda12_1 (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow

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