download_test_data.py 209 B

123456
  1. from datasets import load_dataset
  2. import pandas as pd
  3. dataset = load_dataset("princeton-nlp/SWE-bench")
  4. test = dataset["test"].to_pandas()
  5. test.to_json("data/processed/swe-bench-test.json", orient="records")