1  from data import get_bigcodebench
2  import json
3  
4  subset = 'hard'
5  dataset = get_bigcodebench(subset=subset)
6  
7  save_path = f'bigcodebench_{subset}.json'
8  with open(save_path, 'w') as f:
9      json.dump(dataset, f)