[Feature] Add explicit dimensions and device support to RSSM modules #3305
Dr.CI classification results
{"FAILED":[{"workflowId":20810955530,"workflowUniqueId":55958309,"id":59774832899,"runnerName":"i-0e5b6285f6ac7b2bb","authorEmail":"vincentmoens@gmail.com","name":"Lint / python-source-and-configs / linux-job","jobName":"python-source-and-configs / linux-job","conclusion":"failure","completed_at":"2026-01-08T08:52:51.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20810955530/job/59774832899","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59774832899","head_branch":"gh/vmoens/186/head","pr_number":3305,"head_sha":"1ed37492f6f94d8eea1ef1496dd9cdb919722c01","head_sha_timestamp":"2026-01-08T08:48:30.000000000Z","failure_captures":["RuntimeError: Command docker exec -t de5e75fea36b4092c440f3accfc69832d698cf221e201a34511f3181dbd7ac44 /exec failed with exit code 1"],"failure_lines":["RuntimeError: Command docker exec -t de5e75fea36b4092c440f3accfc69832d698cf221e201a34511f3181dbd7ac44 /exec failed with exit code 1"],"failure_context":[],"time":"2026-01-08T08:48:41.000000000Z"},{"workflowId":20810955518,"workflowUniqueId":60501564,"id":59774834445,"runnerName":"i-09046fe8896a4e793","authorEmail":"vincentmoens@gmail.com","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:00:08.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20810955518/job/59774834445","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59774834445","head_branch":"gh/vmoens/186/head","pr_number":3305,"head_sha":"1ed37492f6f94d8eea1ef1496dd9cdb919722c01","head_sha_timestamp":"2026-01-08T08:48:30.000000000Z","failure_captures":["RuntimeError: Command docker exec -t f55e0cd49d447eab0a8d43f9f372fad799cb02823892144962ded3fdcf03948b /exec failed with exit code 1"],"failure_lines":["RuntimeError: Command docker exec -t f55e0cd49d447eab0a8d43f9f372fad799cb02823892144962ded3fdcf03948b /exec failed with exit code 1"],"failure_context":[],"time":"2026-01-08T08:48:42.000000000Z"},{"workflowId":20810955586,"workflowUniqueId":82165821,"id":59774834911,"runnerName":"i-0bb41e8334dd041e3","authorEmail":"vincentmoens@gmail.com","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:00:48.000000000Z","html_url":"https://github.com/pytorch/rl/actions/runs/20810955586/job/59774834911","logUrl":"https://ossci-raw-job-status.s3.amazonaws.com/log/pytorch/rl/59774834911","head_branch":"gh/vmoens/186/head","pr_number":3305,"head_sha":"1ed37492f6f94d8eea1ef1496dd9cdb919722c01","head_sha_timestamp":"2026-01-08T08:48:30.000000000Z","failure_captures":["RuntimeError: Command docker exec -t f4b4d1a1b4a3318e1e4d7eb1a261f4e3948bbd4a544a1509a11a6b15bf546e6a /exec failed with exit code 1"],"failure_lines":["RuntimeError: Command docker exec -t f4b4d1a1b4a3318e1e4d7eb1a261f4e3948bbd4a544a1509a11a6b15bf546e6a /exec failed with exit code 1"],"failure_context":[],"time":"2026-01-08T08:48:42.000000000Z"}],"FLAKY":[],"BROKEN_TRUNK":[],"UNSTABLE":[]}