Remove use of ObjectID transport flag (#7699)

python/ray/tests/test_failure.py

@@ -113,7 +113,7 @@ def test_get_throws_quickly_when_found_exception(ray_start_regular):
     ray.get(signal1.send.remote())
 
     signal2 = SignalActor.remote()
-    actor = Actor.options(is_direct_call=True, max_concurrency=2).remote()
+    actor = Actor.options(max_concurrency=2).remote()
     expect_exception(
         [actor.bad_func2.remote(),
          actor.slow_func.remote(signal2)], ray.exceptions.RayActorError)

python/ray/tests/test_reference_counting.py

@@ -18,7 +18,14 @@ logger = logging.getLogger(__name__)
 
 @pytest.fixture
 def one_worker_100MiB(request):
-    yield ray.init(num_cpus=1, object_store_memory=100 * 1024 * 1024)
+    config = json.dumps({
+        "object_store_full_max_retries": 2,
+        "task_retry_delay_ms": 0,
+    })
+    yield ray.init(
+        num_cpus=1,
+        object_store_memory=100 * 1024 * 1024,
+        _internal_config=config)
     ray.shutdown()
 
 
@@ -33,12 +40,8 @@ def _fill_object_store_and_get(oid, succeed=True, object_MiB=40,
     if succeed:
         ray.get(oid)
     else:
-        if oid.is_direct_call_type():
-            with pytest.raises(ray.exceptions.RayTimeoutError):
-                ray.get(oid, timeout=0.1)
-        else:
-            with pytest.raises(ray.exceptions.UnreconstructableError):
-                ray.get(oid)
+        with pytest.raises(ray.exceptions.RayTimeoutError):
+            ray.get(oid, timeout=0.1)
 
 
 def _check_refcounts(expected):

python/ray/tests/test_reference_counting_2.py

@@ -18,7 +18,6 @@ logger = logging.getLogger(__name__)
 @pytest.fixture
 def one_worker_100MiB(request):
     config = json.dumps({
-        "distributed_ref_counting_enabled": 1,
         "object_store_full_max_retries": 2,
         "task_retry_delay_ms": 0,
     })
@@ -40,12 +39,8 @@ def _fill_object_store_and_get(oid, succeed=True, object_MiB=40,
     if succeed:
         ray.get(oid)
     else:
-        if oid.is_direct_call_type():
-            with pytest.raises(ray.exceptions.RayTimeoutError):
-                ray.get(oid, timeout=0.1)
-        else:
-            with pytest.raises(ray.exceptions.UnreconstructableError):
-                ray.get(oid)
+        with pytest.raises(ray.exceptions.RayTimeoutError):
+            ray.get(oid, timeout=0.1)
 
 
 # Test that an object containing object IDs within it pins the inner IDs