fix imports

Signed-off-by: Ruiyang Wang <[email protected]>
ray-project · Jan 17, 2025 · 45101dc · 45101dc
1 parent d1c6a2a
commit 45101dc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/python/ray/dashboard/modules/job/job_head.py b/python/ray/dashboard/modules/job/job_head.py
@@ -12,7 +12,6 @@
 
 import ray
 from ray.dashboard.consts import (
-    KV_NAMESPACE_DASHBOARD,
     GCS_RPC_TIMEOUT_SECONDS,
     DASHBOARD_AGENT_ADDR_PREFIX,
     CANDIDATE_AGENT_NUMBER,
@@ -21,7 +20,7 @@
 )
 import ray.dashboard.optional_utils as optional_utils
 import ray.dashboard.utils as dashboard_utils
-from ray._private.ray_constants import env_bool
+from ray._private.ray_constants import env_bool, KV_NAMESPACE_DASHBOARD
 from ray._private.runtime_env.packaging import (
     package_exists,
     pin_runtime_env_uri,

diff --git a/python/ray/dashboard/state_api_utils.py b/python/ray/dashboard/state_api_utils.py
@@ -5,11 +5,10 @@
 import aiohttp.web
 import json
 from ray._private.gcs_utils import GcsAioClient
-from ray.dashboard import dashboard_consts
-from ray._private.ray_constants import KV_NAMESPACE_DASHBOARD, GCS_RPC_TIMEOUT_SECONDS
+from ray.dashboard.consts import GCS_RPC_TIMEOUT_SECONDS, DASHBOARD_AGENT_ADDR_PREFIX
+from ray._private.ray_constants import KV_NAMESPACE_DASHBOARD, GLOBAL_GRPC_OPTIONS
 from ray._private.utils import init_grpc_channel
-from ray._private.runtime_env.constants import GLOBAL_GRPC_OPTIONS
-from ray.core.generated.common_pb2 import NodeID
+from ray import NodeID
 from ray.core.generated.reporter_pb2_grpc import ReporterServiceStub, LogServiceStub
 from typing import Optional
 
@@ -271,7 +270,7 @@ async def get_agent_address(
     If either of them are not found, return None.
     """
     agent_addr_json = await gcs_aio_client.internal_kv_get(
-        f"{dashboard_consts.DASHBOARD_AGENT_ADDR_PREFIX}{node_id.hex()}".encode(),
+        f"{DASHBOARD_AGENT_ADDR_PREFIX}{node_id.hex()}".encode(),
         namespace=KV_NAMESPACE_DASHBOARD,
         timeout=GCS_RPC_TIMEOUT_SECONDS,
     )