(example_dir: str)
| 15 | |
| 16 | |
| 17 | def get_base_example_dir(example_dir: str) -> str: |
| 18 | name = os.path.basename(example_dir) |
| 19 | name = "_".join(name.split("_")[:2]) |
| 20 | return os.path.join("exp_data", "agent_clean", name) |
| 21 | |
| 22 | |
| 23 | def get_examples(traj_root): |