(data)
| 41 | # components.html(html.render(env.sys, qp, height=300), height=300) |
| 42 | |
| 43 | def set_endpoint_to_zero(data): |
| 44 | num_obj = data.shape[-1] // 13 |
| 45 | end_loc = np.zeros([data.shape[0], data.shape[1], num_obj, 3]) |
| 46 | end_loc[..., :18, :2] += np.copy(data[:, -1:, None, :2]) |
| 47 | end_loc[..., 19:, :2] += np.copy(data[:, -1:, None, :2]) |
| 48 | end_loc = end_loc.reshape([data.shape[0], data.shape[1], num_obj * 3]) |
| 49 | data[..., :num_obj*3] = np.copy(data[..., :num_obj*3]) - end_loc |
| 50 | return data |
| 51 | |
| 52 | def set_endpoint_to_xy(data, x, y): |
| 53 | num_obj = data.shape[-1] // 13 |
no outgoing calls
no test coverage detected