fixups

Signed-off-by: Pablo E <pabloem@apache.org>

Author: pabloem

python/ray/_raylet.pyx

@@ -200,6 +200,8 @@ cdef int check_status(const CRayStatus& status) nogil except -1:
         raise ValueError(message)
     elif status.IsObjectNotFound():
         raise ValueError(message)
+    elif status.IsObjectUnknownOwner():
+        raise ValueError(message)
     else:
         raise RaySystemError(message)
 

python/ray/includes/common.pxd

@@ -115,6 +115,7 @@ cdef extern from "ray/common/status.h" namespace "ray" nogil:
         c_bool ShouldExitWorker()
         c_bool IsObjectNotFound()
         c_bool IsNotFound()
+        c_bool IsObjectUnknownOwner()
 
         c_string ToString()
         c_string CodeAsString()

python/ray/tests/test_ray_shutdown.py

@@ -70,7 +70,7 @@ def f(s):
     ray.shutdown()
 
     ray.init(num_cpus=1, include_dashboard=False)
-    with pytest.raises(ValueError, match="a Ray object whose owner is unknown"):
+    with pytest.raises(ValueError, match="Ray object whose owner is unknown"):
         f.remote(my_ref)  # This would cause full CPython death.
 
     ray.shutdown()
@@ -92,7 +92,7 @@ def f(s):
     ray.shutdown()
 
     ray.init(num_cpus=1, include_dashboard=False)
-    with pytest.raises(ValueError, match="a Ray object whose owner is unknown"):
+    with pytest.raises(ValueError, match="Ray object whose owner is unknown"):
         f.remote([my_ref])  # This would cause full CPython death.
 
     ray.shutdown()
@@ -110,7 +110,7 @@ def test_ray_shutdown_then_get(short_gcs_publish_timeout, shutdown_only):
     ray.shutdown()
 
     ray.init(num_cpus=1, include_dashboard=False)
-    with pytest.raises(ValueError, match="a Ray object whose owner is unknown"):
+    with pytest.raises(ValueError, match="Ray objects whose owner is unknown"):
         # This used to cause ray to hang indefinitely (without timeout) or
         # throw a timeout exception if a timeout was provided. Now it is expected to
         # throw an exception reporting the unknown object.
@@ -137,7 +137,7 @@ def test_ray_shutdown_then_wait(short_gcs_publish_timeout, shutdown_only):
     # function to wait for the valid references; however, if all the
     # references are unknown, we expect an error.
     ready, not_ready = ray.wait([my_new_ref, my_ref])
-    with pytest.raises(ValueError, match="a Ray object whose owner is unknown"):
+    with pytest.raises(ValueError, match="Ray object whose owner is unknown"):
         # This used to cause ray to hang indefinitely (without timeout) or
         # forever return all tasks as not-ready if a timeout was provided.
         # Now it is expected to throw an exception reporting if all objects are

src/ray/common/status.h

@@ -269,6 +269,7 @@ class RAY_EXPORT Status {
   bool IsSchedulingCancelled() const { return code() == StatusCode::SchedulingCancelled; }
   bool IsObjectExists() const { return code() == StatusCode::ObjectExists; }
   bool IsObjectNotFound() const { return code() == StatusCode::ObjectNotFound; }
+  bool IsObjectUnknownOwner() const { return code() == StatusCode::ObjectUnknownOwner; }
   bool IsObjectAlreadySealed() const { return code() == StatusCode::ObjectAlreadySealed; }
   bool IsObjectStoreFull() const { return code() == StatusCode::ObjectStoreFull; }
   bool IsTransientObjectStoreFull() const {