from datasets import load_dataset
dataset = load_dataset("YifeiDevs/scatter-plot", split="train")
print(f"Loaded {len(dataset)} samples. Fields: {dataset.column_names}")
sample = dataset[0]
print(f"\nImage: {sample['image'].size} {sample['image'].mode}")
print(f"User: {sample['user']}")
print(f"Thinking: {sample['thinking'][:200]}...")
print(f"Assistant: {sample['assistant'][:200]}...")
