ray.rllib.evaluation.rollout_worker.RolloutWorker.set_policy_mapping_fn

RolloutWorker.set_policy_mapping_fn(policy_mapping_fn: Optional[Callable[[Any, Episode], str]] = None) → None

Sets self.policy_mapping_fn to a new callable (if provided).

Parameters

policy_mapping_fn – The new mapping function to use. If None, will keep the existing mapping function in place.