brysonjones/pusht_keypoints

{
    "codebase_version": "v2.0",
    "robot_type": "unknown",
    "total_episodes": 206,
    "total_frames": 25650,
    "total_tasks": 1,
    "total_videos": 0,
    "total_chunks": 1,
    "chunks_size": 1000,
    "fps": 10,
    "splits": {
        "train": "0:206"
    },
    "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet",
    "video_path": null,
    "features": {
        "observation.state": {
            "dtype": "float32",
            "shape": [
                2
            ],
            "names": {
                "motors": [
                    "motor_0",
                    "motor_1"
                ]
            }
        },
        "observation.environment_state": {
            "dtype": "float32",
            "shape": [
                16
            ],
            "names": {
                "motors": [
                    "motor_0",
                    "motor_1",
                    "motor_2",
                    "motor_3",
                    "motor_4",
                    "motor_5",
                    "motor_6",
                    "motor_7",
                    "motor_8",
                    "motor_9",
                    "motor_10",
                    "motor_11",
                    "motor_12",
                    "motor_13",
                    "motor_14",
                    "motor_15"
                ]
            }
        },
        "action": {
            "dtype": "float32",
            "shape": [
                2
            ],
            "names": {
                "motors": [
                    "motor_0",
                    "motor_1"
                ]
            }
        },
        "episode_index": {
            "dtype": "int64",
            "shape": [
                1
            ],
            "names": null
        },
        "frame_index": {
            "dtype": "int64",
            "shape": [
                1
            ],
            "names": null
        },
        "timestamp": {
            "dtype": "float32",
            "shape": [
                1
            ],
            "names": null
        },
        "next.reward": {
            "dtype": "float32",
            "shape": [
                1
            ],
            "names": null
        },
        "next.done": {
            "dtype": "bool",
            "shape": [
                1
            ],
            "names": null
        },
        "next.success": {
            "dtype": "bool",
            "shape": [
                1
            ],
            "names": null
        },
        "index": {
            "dtype": "int64",
            "shape": [
                1
            ],
            "names": null
        },
        "task_index": {
            "dtype": "int64",
            "shape": [
                1
            ],
            "names": null
        }
    }
}