ray.rllib.utils.checkpoints.convert_to_msgpack_policy_checkpoint#

ray.rllib.utils.checkpoints.convert_to_msgpack_policy_checkpoint(policy_checkpoint: str | Checkpoint, msgpack_checkpoint_dir: str) → str[source]#

Converts a Policy checkpoint (pickle based) to a msgpack based one.

Msgpack has the advantage of being python version independent.

Parameters:

  • policy_checkpoint – The directory, in which to find the Policy checkpoint (pickle based).

  • msgpack_checkpoint_dir – The directory, in which to create the new msgpack based checkpoint.

Returns:

The directory in which the msgpack checkpoint has been created. Note that this is the same as msgpack_checkpoint_dir.

PublicAPI (beta): This API is in beta and may change before becoming stable.