["size_categories:1K<n<10K", "format:parquet", "library:datasets", "library:pandas", "library:mlcroissant", "library:polars", "region:us"]
[{"name": "model_inputs", "struct": [{"name": "preorder_index", "sequence": "int32"}, {"name": "postorder_index", "sequence": "int32"}, {"name": "input_ids", "sequence": "int32"}, {"name": "labels", "sequence": "int32"}, {"name": "position_ids", "sequence": "int32"}, {"name": "sequence_id", "sequence": "int32"}]}, {"name": "loss_inputs", "struct": [{"name": "parent_index", "sequence": "int32"}, {"name": "num_correct", "sequence": "int32"}, {"name": "num_rollouts", "sequence": "int32"}, {"name": "parent_rollout_index", "sequence": "int32"}]}]