[Core] Expand job ID to 4 bytes by removing object flag bytes. (#10187)

python/ray/includes/unique_ids.pxd

@@ -139,11 +139,7 @@ cdef extern from "ray/common/id.h" namespace "ray" nogil:
         const CObjectID Nil()
 
         @staticmethod
-        CObjectID ForPut(const CTaskID &task_id, int64_t index,
-                         int64_t transport_type)
-
-        @staticmethod
-        CObjectID ForTaskReturn(const CTaskID &task_id, int64_t index)
+        CObjectID FromIndex(const CTaskID &task_id, int64_t index)
 
         @staticmethod
         size_t Size()

python/ray/includes/unique_ids.pxi

@@ -221,7 +221,7 @@ cdef class JobID(BaseID):
 
     @classmethod
     def from_int(cls, value):
-        assert value < 65536, "Maximum JobID integer is 65535."
+        assert value < 2**32, "Maximum JobID integer is 2**32 - 1."
         return cls(CJobID.FromInt(value).Binary())
 
     @classmethod