Feature Request: Native Dict Observation Space Support and Improved Stacking in StackedObservations
Feature Request: Native Dict Observation Space Support and Improved Stacking in StackedObservations: a task in LegoFlow-SWE (Harbor dataset). The StackedObservations class currently only handles gym.spaces.Box observation spaces. Users who need to stack observations from environments with…
The task
The `StackedObservations` class currently only handles `gym.spaces.Box` observation spaces. Users who need to stack observations from environments with `gym.spaces.Dict` observation spaces must rely on a separate, non-unified implementation. This leads to code duplication and an inconsistent API.
Part of Lego-X/LegoFlow-SWE.