Add channel dimension to all environment observation spaces

david-lindner / safe-grid-gym

A gym interface for AI safety gridworlds created in pycolab.

Apache License 2.0

17 stars 9 forks source link

Add channel dimension to all environment observation spaces #23

Closed jvmncs closed 5 years ago

jvmncs commented 5 years ago

When use_transitions=True, we end up with observation tensors of rank 3. When False, we were seeing observation tensors of rank 2. This PR normalizes all observation tensors to rank 3 for consistency.