ray.rllib.evaluation.rollout_worker.RolloutWorker.set_policy_mapping_fn#

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

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.