(train_json_path)
| 235 | return test_data |
| 236 | |
| 237 | def read_train_list(train_json_path): |
| 238 | try: |
| 239 | with open(train_json_path, 'r', encoding='utf-8') as file: |
| 240 | data = json.load(file) |
| 241 | return set(data.get('train', [])) |
| 242 | except Exception as e: |
| 243 | print(f"Error at: {str(e)}") |
| 244 | return set() |
| 245 | |
| 246 | def read_episodes_from_jsonl(jsonl_file_path, train_episodes, output_test_dir): |
| 247 | episodes = {'in_train': [], 'not_in_train': []} |