def testObjRefCheckpointSerde(self):
# Obj ref checkpoints are dict checkpoints put into the Ray object
# store, but they have their own data representation (the obj ref).
# We thus compare with the actual obj ref checkpoint.
source_checkpoint = Checkpoint.from_dict({"checkpoint_data": 5})
obj_ref = source_checkpoint.to_object_ref()
checkpoint = Checkpoint.from_object_ref(obj_ref)
self._testCheckpointSerde(checkpoint, *checkpoint.get_internal_representation())
def test_fs_checkpoint_obj_store(self):
"""Test conversion from fs to obj store checkpoint and back."""
if not ray.is_initialized():
ray.init()
checkpoint = self._prepare_fs_checkpoint()
# Convert into obj ref checkpoint
obj_ref = checkpoint.to_object_ref()
# Create from object ref
checkpoint = Checkpoint.from_object_ref(obj_ref)
self.assertIsInstance(checkpoint._obj_ref, ray.ObjectRef)
self._assert_fs_checkpoint(checkpoint)
def __init__(self):
checkpoint = Checkpoint.from_object_ref(checkpoint_object_ref)
self.predictor = TorchPredictor.from_checkpoint(checkpoint)
def __init__(self):
self.predictor = LightGBMPredictor.from_checkpoint(
Checkpoint.from_object_ref(checkpoint_object_ref)
)
def __init__(self):
self.predictor = TensorflowPredictor.from_checkpoint(
Checkpoint.from_object_ref(checkpoint_object_ref),
model_definition=build_model,
)
def __init__(self):
self.predictor = XGBoostPredictor.from_checkpoint(
Checkpoint.from_object_ref(checkpoint_object_ref))
