split_indices = json.load(open(os.path.join(base_dir, "pid_splits.json")))[split] problems = json.load(open(os.path.join(base_dir, "problems.json"))) ...