Skip to content

[Feature] Add _Contiguous module and reshape improvements to encoders/decoders #3306

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
vmoens wants to merge 4 commits into
base: gh/vmoens/187/base
Choose a base branch
from

Update

348c94d
Select commit
Loading
Failed to load commit list.
Open

[Feature] Add _Contiguous module and reshape improvements to encoders/decoders #3306

Update
348c94d
Select commit
Loading
Failed to load commit list.
PyTorch Bot / Dr.CI completed Jan 8, 2026 in 0s

Dr.CI classification results

{"FAILED":[{"workflowId":20811374989,"workflowUniqueId":61790681,"id":59776274114,"runnerName":"i-0e99387fd13c2cd01","authorEmail":"[email protected]","name":"Continuous Benchmark (PR) / CPU Pytest benchmark","jobName":"CPU Pytest benchmark","conclusion":"failure","completed_at":"2026-01-08T09:38:55.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811374989/job/59776274114","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776274114","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["Process completed with exit code 1."],"failure_lines":["##[error]Process completed with exit code 1."],"failure_context":[],"time":"2026-01-08T09:05:19.000000000Z"},{"workflowId":20811374989,"workflowUniqueId":61790681,"id":59776274162,"runnerName":"i-0280bb7a7272ae194","authorEmail":"[email protected]","name":"Continuous Benchmark (PR) / GPU Pytest benchmark","jobName":"GPU Pytest benchmark","conclusion":"cancelled","completed_at":"2026-01-08T09:39:31.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811374989/job/59776274162","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776274162","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":[],"failure_lines":[],"failure_context":[],"time":"2026-01-08T09:05:19.000000000Z"},{"workflowId":20811375094,"workflowUniqueId":82165821,"id":59776223700,"runnerName":"i-0371431f93c906688","authorEmail":"[email protected]","name":"Libs Tests on Linux / unittests-gym (3.9, 12.8) / linux-job","jobName":"unittests-gym (3.9, 12.8) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:15:11.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375094/job/59776223700","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776223700","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["RuntimeError: Command docker exec -t 727ca317c5b7a32bad25900879bdb35d99a98ab202ee19473cd4759165946bde /exec failed with exit code 1"],"failure_lines":["RuntimeError: Command docker exec -t 727ca317c5b7a32bad25900879bdb35d99a98ab202ee19473cd4759165946bde /exec failed with exit code 1"],"failure_context":[],"time":"2026-01-08T09:04:43.000000000Z"},{"workflowId":20811375106,"workflowUniqueId":128761754,"id":59776319763,"runnerName":"i-047a6797e660cba80","authorEmail":"[email protected]","name":"SOTA Tests on Linux / tests (3.10, 13.0) / linux-job","jobName":"tests (3.10, 13.0) / linux-job","conclusion":"cancelled","completed_at":"2026-01-08T10:36:59.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375106/job/59776319763","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776319763","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["##[error]The operation was canceled."],"failure_lines":["##[error]The operation was canceled."],"failure_context":[],"time":"2026-01-08T09:05:50.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322882,"runnerName":"i-0ab3be829023a6084","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-cpu (3.12) / linux-job","jobName":"tests-cpu (3.12) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:55:48.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322882","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322882","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322912,"runnerName":"i-0ecad7fa9ca12635e","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-gpu-distributed (3.12, 13.0) / linux-job","jobName":"tests-gpu-distributed (3.12, 13.0) / linux-job","conclusion":"failure","completed_at":"2026-01-08T10:07:25.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322912","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322912","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor"],"failure_lines":["FAILED test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/3 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322913,"runnerName":"i-0f3deb4ed92dd2c22","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-cpu (3.11) / linux-job","jobName":"tests-cpu (3.11) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:53:14.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322913","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322913","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322915,"runnerName":"i-0a8e9f5fd7a2dd2aa","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-stable-gpu (3.12, 13.0) / linux-job","jobName":"tests-stable-gpu (3.12, 13.0) / linux-job","conclusion":"failure","completed_at":"2026-01-08T10:37:00.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322915","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322915","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322932,"runnerName":"i-088f1bce1bb910963","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-cpu (3.13) / linux-job","jobName":"tests-cpu (3.13) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:52:55.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322932","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322932","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322888,"runnerName":"i-0cd70cd01d8fb1a56","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-olddeps (3.9, 11.6) / linux-job","jobName":"tests-olddeps (3.9, 11.6) / linux-job","conclusion":"failure","completed_at":"2026-01-08T10:19:06.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322888","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322888","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - RuntimeError: Socket Timeout"],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322889,"runnerName":"i-04ae5e15baadc1551","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-stable-gpu-distributed (3.12, 13.0) / linux-job","jobName":"tests-stable-gpu-distributed (3.12, 13.0) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:55:42.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322889","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322889","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor"],"failure_lines":["FAILED test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/3 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322891,"runnerName":"i-0661e422651bddfc2","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-gpu (3.12, 13.0) / linux-job","jobName":"tests-gpu (3.12, 13.0) / linux-job","conclusion":"failure","completed_at":"2026-01-08T10:36:54.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322891","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322891","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322900,"runnerName":"i-08fc8b181e1642bfa","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-optdeps (3.12, 13.0) / linux-job","jobName":"tests-optdeps (3.12, 13.0) / linux-job","conclusion":"failure","completed_at":"2026-01-08T10:16:37.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322900","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322900","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322901,"runnerName":"i-00c5aef371ce2a9a5","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-cpu (3.10) / linux-job","jobName":"tests-cpu (3.10) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:54:20.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322901","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322901","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector"],"failure_lines":["FAILED test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector - torch.distributed.DistStoreError: Timed out after 61 seconds waiting for clients. 1/2 clients joined."],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375126,"workflowUniqueId":60501564,"id":59776222938,"runnerName":"i-0ca9f9126dc977d7a","authorEmail":"[email protected]","name":"Habitat Tests on Linux / tests (3.9, 12.8) / linux-job","jobName":"tests (3.9, 12.8) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:14:53.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375126/job/59776222938","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776222938","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["RuntimeError: Command docker exec -t cd1439ad8881b565a23f7b52e1037471b430f2f76d404680c964b6a0a2b2d0b2 /exec failed with exit code 1"],"failure_lines":["RuntimeError: Command docker exec -t cd1439ad8881b565a23f7b52e1037471b430f2f76d404680c964b6a0a2b2d0b2 /exec failed with exit code 1"],"failure_context":[],"time":"2026-01-08T09:04:43.000000000Z"}],"FLAKY":[],"BROKEN_TRUNK":[{"workflowId":20811375113,"workflowUniqueId":82165822,"id":59776322902,"runnerName":"i-05d0967049b3e6462","authorEmail":"[email protected]","name":"Unit-tests on Linux / tests-cpu (3.14) / linux-job","jobName":"tests-cpu (3.14) / linux-job","conclusion":"failure","completed_at":"2026-01-08T09:39:54.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375113/job/59776322902","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776322902","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False]"],"failure_lines":["FAILED test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False] - AttributeError: 'NonTensorStack' object has no attribute '_td_dim_names'. Did you mean: '_td_dim_name'?"],"failure_context":[],"time":"2026-01-08T09:05:52.000000000Z"},{"workflowId":20811375103,"workflowUniqueId":79519303,"id":59776244446,"runnerName":"i-07516e71cce594d08","authorEmail":"[email protected]","name":"Unit-tests on Windows / unittests-cpu (3.10, windows.4xlarge, cpu) / windows-job","jobName":"unittests-cpu (3.10, windows.4xlarge, cpu) / windows-job","conclusion":"failure","completed_at":"2026-01-08T09:51:28.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20811375103/job/59776244446","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59776244446","head_branch":"gh/vmoens/187/head","pr_number":3306,"head_sha":"348c94d3f7796a84be98538660dfde50d96e6150","head_sha_timestamp":"2026-01-08T09:04:34.000000000Z","failure_captures":["test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False]"],"failure_lines":["FAILED test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False] - AttributeError: 'NonTensorStack' object has no attribute '_td_dim_names'. Did you mean: '_td_dim_name'?"],"failure_context":[],"time":"2026-01-08T09:04:58.000000000Z"}],"UNSTABLE":[]}

View more details on PyTorch Bot