Source code for ray.experimental.object_spilling

import ray


[docs]def force_spill_objects(object_refs): """Force spilling objects to external storage. Args: object_refs: Object refs of the objects to be spilled. """ core_worker = ray.worker.global_worker.core_worker # Make sure that the values are object refs. for object_ref in object_refs: if not isinstance(object_ref, ray.ObjectRef): raise TypeError( f"Attempting to call `force_spill_objects` on the " f"value {object_ref}, which is not an ray.ObjectRef.") return core_worker.force_spill_objects(object_refs)
[docs]def force_restore_spilled_objects(object_refs): """Force restoring objects from external storage. Args: object_refs: Object refs of the objects to be restored. """ core_worker = ray.worker.global_worker.core_worker # Make sure that the values are object refs. for object_ref in object_refs: if not isinstance(object_ref, ray.ObjectRef): raise TypeError( f"Attempting to call `force_restore_spilled_objects` on the " f"value {object_ref}, which is not an ray.ObjectRef.") return core_worker.force_restore_spilled_objects(object_refs)