hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Dashboard] [Serve] Make serve import conditional (#17713)

Browse source
This commit is contained in:
architkulkarni 2021-08-10 17:06:00 -07:00 • committed by GitHub
parent 63c15d7ced
commit 0c2c99b951
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
dashboard/modules/snapshot/snapshot_head.py
View file

@ -4,9 +4,6 @@ from ray.core.generated import gcs_service_pb2_grpc
from ray.experimental.internal_kv import _internal_kv_get
import ray.new_dashboard.utils as dashboard_utils
from ray.serve.controller import SNAPSHOT_KEY as SERVE_SNAPSHOT_KEY
from ray.serve.constants import SERVE_CONTROLLER_NAME
from ray.serve.kv_store import format_key
import json
@ -91,6 +88,15 @@ class SnapshotHead(dashboard_utils.DashboardHeadModule):
return actors
async def get_serve_info(self):
# Conditionally import serve to prevent ModuleNotFoundError from serve
# dependencies when only ray[default] is installed (#17712)
try:
from ray.serve.controller import SNAPSHOT_KEY as SERVE_SNAPSHOT_KEY
from ray.serve.constants import SERVE_CONTROLLER_NAME
from ray.serve.kv_store import format_key
except Exception:
return "{}"
client = self._dashboard_head.gcs_client
# Serve wraps Ray's internal KV store and specially formats the keys.