MCPcopy Create free account
hub / github.com/Speakn0w/RealChart2Code / find_final_data_files

Function find_final_data_files

RealChart2Code_eval/evaluate_task2.py:67–90  ·  view source on GitHub ↗
(data_dir: Path, difficulty: str = None)

Source from the content-addressed store, hash-verified

65
66
67def find_final_data_files(data_dir: Path, difficulty: str = None) -> List[Path]:
68
69 data_files = []
70
71 for dir_path in data_dir.iterdir():
72 if not dir_path.is_dir():
73 continue
74
75 if not any(dir_path.glob("task_*.json")):
76 continue
77
78 difficulties = [difficulty] if difficulty else ["simple", "middle", "hard"]
79
80 for diff in difficulties:
81
82 standard_task = dir_path / f"task_{diff}_data.json"
83 if standard_task.exists():
84 data_files.append(standard_task)
85
86 multi_task = dir_path / f"task_{diff}_mul_data.json"
87 if multi_task.exists():
88 data_files.append(multi_task)
89
90 return data_files
91
92def parse_task_file(task_file: Path) -> Tuple[str, str, List[str]]:
93

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected