diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx index 6b15ba0dc..439f079f8 100644 --- a/python/ray/_raylet.pyx +++ b/python/ray/_raylet.pyx @@ -1726,7 +1726,6 @@ cdef class CoreWorker: CObjectID c_object_id = object_ref.native() CAddress c_owner_address = CAddress() c_string serialized_object_status - CCoreWorkerProcess.GetCoreWorker().PromoteObjectToPlasma(c_object_id) CCoreWorkerProcess.GetCoreWorker().GetOwnershipInfo( c_object_id, &c_owner_address, &serialized_object_status) return (object_ref, diff --git a/src/ray/core_worker/core_worker.cc b/src/ray/core_worker/core_worker.cc index e28f2c5c0..15540055a 100644 --- a/src/ray/core_worker/core_worker.cc +++ b/src/ray/core_worker/core_worker.cc @@ -1025,6 +1025,7 @@ void CoreWorker::PutObjectIntoPlasma(const RayObject &object, const ObjectID &ob } void CoreWorker::PromoteObjectToPlasma(const ObjectID &object_id) { + // TODO(swang): Remove. auto value = memory_store_->GetOrPromoteToPlasma(object_id); if (value) { PutObjectIntoPlasma(*value, object_id);