update comment and move import to top-level

Signed-off-by: kaihsun <kaihsun@anyscale.com>
ray-project · kevin85421 · Dec 26, 2024 · Dec 26, 2024 · Dec 26, 2024 · Dec 26, 2024
commit d1aac6c15b24adaf3a0c619af0aa61afb6ae31c8
diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx
@@ -3437,7 +3437,7 @@ cdef class CoreWorker:
                          .WaitAndGetExperimentalMutableObjects(
                             c_object_ids, timeout_ms, num_returns, results))
 
-        # Users can determine whether `timeout` was raised by checking the value
+        # The caller can determine whether `timeout` was raised by checking the value
         # of `results`. At the same time, users still want to get the values of some
         # objects even if some objects are not ready. Hence, we don't raise
         # the exception if `suppress_timeout_errors` is set to True and instead return

diff --git a/python/ray/experimental/channel/shared_memory_channel.py b/python/ray/experimental/channel/shared_memory_channel.py
@@ -8,7 +8,7 @@
 import ray.exceptions
 from ray import ObjectRef
 from ray._raylet import SerializedObject
-from ray.experimental.channel import utils
+from ray.experimental.channel import ChannelContext, utils
 from ray.experimental.channel.common import ChannelInterface, ChannelOutputType
 from ray.experimental.channel.intra_process_channel import IntraProcessChannel
 from ray.experimental.channel.utils import get_self_actor
@@ -477,8 +477,6 @@ def read(self, timeout: Optional[float] = None) -> Any:
 
         start_time = time.monotonic()
 
-        from ray.experimental.channel import ChannelContext
-
         ctx = ChannelContext.get_current().serialization_context
         ret = ctx.get_data(self._local_reader_ref)