{
  "action": {
    "type": "ACTION",
    "shape": [
      20
    ]
  },
  "observation.image.0": {
    "type": "VISUAL",
    "shape": [
      3,
      256,
      256
    ]
  },
  "observation.image.1": {
    "type": "VISUAL",
    "shape": [
      3,
      256,
      256
    ]
  },
  "observation.image.2": {
    "type": "VISUAL",
    "shape": [
      3,
      256,
      256
    ]
  },
  "observation.state": {
    "type": "STATE",
    "shape": [
      20
    ]
  }
}