| from sim_reveal.dataset import RevealOfflineDataset, collect_teacher_dataset | |
| def test_no_leak_with_new_labels(): | |
| bundle = collect_teacher_dataset( | |
| proxy_names=["bag_proxy"], | |
| episodes_per_proxy=1, | |
| resolution=32, | |
| seed=123, | |
| chunk_horizon=4, | |
| rollout_horizon=3, | |
| history_steps=2, | |
| planner_candidates=4, | |
| dataset_version="reveal_proxy_v6_rgbd_elastic_state", | |
| ) | |
| dataset = RevealOfflineDataset(bundle["samples"], resolution=32) | |
| item = dataset[0] | |
| for key in ( | |
| "gap_width", | |
| "hold_quality", | |
| "mouth_aperture", | |
| "fold_preservation", | |
| "lift_too_much_risk", | |
| ): | |
| assert key in item | |
| assert "gap_width" not in item["texts"].lower() | |