Fix

Signed-off-by: SangBin Cho <[email protected]>

python/ray/_private/node.py

@@ -203,13 +203,6 @@ def __init__(
  node_ip_address=node_ip_address, raylet_ip_address=node_ip_address
  )
 
- # Try to get node IP address with the parameters.
- # if ray_params.node_ip_address:
- # node_ip_address = ray_params.node_ip_address
- # elif ray_params.redis_address:
- # node_ip_address = ray.util.get_node_ip_address(ray_params.redis_address)
- # else:
- # node_ip_address = ray.util.get_node_ip_address()
  self._node_ip_address = node_ip_address
 
  if ray_params.raylet_ip_address:

python/ray/_private/worker.py

@@ -1284,6 +1284,7 @@ def init(
  )
  _redis_max_memory: Optional[int] = kwargs.pop("_redis_max_memory", None)
  _plasma_directory: Optional[str] = kwargs.pop("_plasma_directory", None)
+ _node_ip_address: str = kwargs.pop("_node_ip_address", None)
  _driver_object_store_memory: Optional[int] = kwargs.pop(
  "_driver_object_store_memory", None
  )
@@ -1471,6 +1472,7 @@ def init(
 
  # Use a random port by not specifying Redis port / GCS server port.
  ray_params = ray._private.parameter.RayParams(
+ node_ip_address=_node_ip_address,
  object_ref_seed=None,
  driver_mode=driver_mode,
  redirect_output=None,
@@ -1553,6 +1555,7 @@ def init(
 
  # In this case, we only need to connect the node.
  ray_params = ray._private.parameter.RayParams(
+ node_ip_address=_node_ip_address,
  gcs_address=gcs_address,
  redis_address=redis_address,
  redis_password=_redis_password,