def decode_params(env_observation: np.ndarray, arguments: IntegerArguments): return env_observation.argmax(axis=1), arguments.decode_all()