lerobot/tests/data/xarm_lift_medium/train/dataset_info.json

{
  "citation": "",
  "description": "",
  "features": {
    "observation.image": {
      "_type": "Image"
    },
    "observation.state": {
      "feature": {
        "dtype": "float32",
        "_type": "Value"
      },
      "length": 4,
      "_type": "Sequence"
    },
    "action": {
      "feature": {
        "dtype": "float32",
        "_type": "Value"
      },
      "length": 4,
      "_type": "Sequence"
    },
    "episode_index": {
      "dtype": "int64",
      "_type": "Value"
    },
    "frame_index": {
      "dtype": "int64",
      "_type": "Value"
    },
    "timestamp": {
      "dtype": "float32",
      "_type": "Value"
    },
    "next.reward": {
      "dtype": "float32",
      "_type": "Value"
    },
    "next.done": {
      "dtype": "bool",
      "_type": "Value"
    },
    "index": {
      "dtype": "int64",
      "_type": "Value"
    }
  },
  "homepage": "",
  "license": ""
}