diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx index b287e60c4..a5a6981d7 100644 --- a/python/ray/_raylet.pyx +++ b/python/ray/_raylet.pyx @@ -354,7 +354,7 @@ cdef class RayletClient: cdef CActorCheckpointID checkpoint_id check_status(self.client.get().PrepareActorCheckpoint( actor_id.data, checkpoint_id)) - return ObjectID.from_native(checkpoint_id) + return ActorCheckpointID.from_native(checkpoint_id) def notify_actor_resumed_from_checkpoint(self, ActorID actor_id, ActorCheckpointID checkpoint_id): diff --git a/python/ray/includes/unique_ids.pxi b/python/ray/includes/unique_ids.pxi index 358d67b3e..9774c285b 100644 --- a/python/ray/includes/unique_ids.pxi +++ b/python/ray/includes/unique_ids.pxi @@ -74,7 +74,7 @@ cdef class UniqueID: return self.data.is_nil() def __eq__(self, other): - return self.binary() == other.binary() + return type(self) == type(other) and self.binary() == other.binary() def __ne__(self, other): return self.binary() != other.binary() @@ -263,7 +263,7 @@ cdef class ActorCheckpointID(UniqueID): @staticmethod cdef from_native(const CActorCheckpointID& cpp_id): - cdef ActorCheckpointID self = ActorCheckpointID.__new__(ActorHandleID) + cdef ActorCheckpointID self = ActorCheckpointID.__new__(ActorCheckpointID) self.data = cpp_id return self