(theta: np.ndarray)
| 78 | return np.concatenate([game, mu]) |
| 79 | |
| 80 | def unpack(theta: np.ndarray) -> tuple[np.ndarray, np.ndarray]: |
| 81 | game = theta[:p] |
| 82 | mu = theta[p:] |
| 83 | return game, mu |
| 84 | |
| 85 | valid_mask = np.ones((T, R), dtype=bool) |
| 86 | valid_mask[:, :p] = False |