(qp)
| 7 | return jp.tree_map(lambda x, y: np.concatenate([x,y], axis=-2), qp, box_qp) |
| 8 | |
| 9 | def remove_box(qp): |
| 10 | return jp.tree_map(lambda x: x[..., :19, :], qp), jp.tree_map(lambda x: x[..., 19:, :], qp) |
| 11 | |
| 12 | def add_scene_to_traj(traj, waypoint, scene='hit', bar_height=1.25): |
| 13 | if scene=='hit': |
no outgoing calls
no test coverage detected