[Autoscaler] Remove duplicate util function is_head_node (ray-proje…

…ct#50615) The following two functions are the same. https://github.com/ray-project/ray/blob/30df6a43a7324eaf65b83b94b5c1511b0f954a39/python/ray/autoscaler/v2/utils.py#L840-L851 https://github.com/ray-project/ray/blob/30df6a43a7324eaf65b83b94b5c1511b0f954a39/python/ray/autoscaler/v2/sdk.py#L100-L110 --------- Signed-off-by: kaihsun <[email protected]>
hcc429 · Feb 16, 2025 · 4976d0e · 4976d0e
1 parent 52d5eb9
commit 4976d0e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 15 deletions.
diff --git a/python/ray/autoscaler/v2/instance_manager/reconciler.py b/python/ray/autoscaler/v2/instance_manager/reconciler.py
@@ -26,7 +26,7 @@
 from ray.autoscaler.v2.metrics_reporter import AutoscalerMetricsReporter
 from ray.autoscaler.v2.scheduler import IResourceScheduler, SchedulingRequest
 from ray.autoscaler.v2.schema import AutoscalerInstance, NodeType
-from ray.autoscaler.v2.sdk import is_head_node
+from ray.autoscaler.v2.utils import is_head_node
 from ray.core.generated.autoscaler_pb2 import (
     AutoscalingState,
     ClusterResourceState,

diff --git a/python/ray/autoscaler/v2/sdk.py b/python/ray/autoscaler/v2/sdk.py
@@ -9,7 +9,6 @@
     ClusterResourceState,
     GetClusterResourceStateReply,
     GetClusterStatusReply,
-    NodeState,
 )
 
 DEFAULT_RPC_TIMEOUT_S = 10
@@ -95,16 +94,3 @@ def get_cluster_resource_state(gcs_client: GcsClient) -> ClusterResourceState:
     reply = GetClusterResourceStateReply()
     reply.ParseFromString(str_reply)
     return reply.cluster_resource_state
-
-
-def is_head_node(node_state: NodeState) -> bool:
-    """
-    Check if the node is a head node from the node state.
-    Args:
-        node_state: the node state
-    Returns:
-        is_head: True if the node is a head node, False otherwise.
-    """
-    # TODO: we should include this bit of information in the future. e.g.
-    # from labels.
-    return "node:__internal_head__" in dict(node_state.total_resources)