| 118 | |
| 119 | # save data |
| 120 | def dump_traj(traj, out_root: Path, idx: int): |
| 121 | if not traj: |
| 122 | return |
| 123 | |
| 124 | subfolder_name = f"traj_{idx:05d}" |
| 125 | subfolder_path = out_root / subfolder_name |
| 126 | subfolder_path.mkdir(parents=True, exist_ok=True) |
| 127 | |
| 128 | out_filename = f"{subfolder_name}.json" |
| 129 | out_file = subfolder_path / out_filename |
| 130 | |
| 131 | with out_file.open("w", encoding="utf-8") as fw: |
| 132 | json.dump(traj, fw, ensure_ascii=False, indent=2) |
| 133 | |
| 134 | print(f"Save {subfolder_name}/{out_filename} (steps={len(traj)})") |
| 135 | |
| 136 | |
| 137 | records = data_eval |