| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- from typing import Any, Dict, List, Optional, Tuple
- from ray.dag import DAGNode
- from ray.dag.format_utils import get_dag_node_str
- from ray.serve.deployment import Deployment
- from ray.serve.handle import DeploymentHandle
- class DeploymentNode(DAGNode):
- """Represents a deployment node in a DAG authored Ray DAG API."""
- def __init__(
- self,
- # For serve structured deployment, deployment body can be import path
- # to the class or function instead.
- deployment: Deployment,
- app_name: str,
- deployment_init_args: Tuple[Any],
- deployment_init_kwargs: Dict[str, Any],
- ray_actor_options: Dict[str, Any],
- other_args_to_resolve: Optional[Dict[str, Any]] = None,
- ):
- # Assign instance variables in base class constructor.
- super().__init__(
- deployment_init_args,
- deployment_init_kwargs,
- ray_actor_options,
- other_args_to_resolve=other_args_to_resolve,
- )
- self._app_name = app_name
- self._deployment = deployment
- self._deployment_handle = DeploymentHandle(
- self._deployment.name, self._app_name
- )
- def _copy_impl(
- self,
- new_args: List[Any],
- new_kwargs: Dict[str, Any],
- new_options: Dict[str, Any],
- new_other_args_to_resolve: Dict[str, Any],
- ):
- return DeploymentNode(
- self._deployment,
- self._app_name,
- new_args,
- new_kwargs,
- new_options,
- other_args_to_resolve=new_other_args_to_resolve,
- )
- def __str__(self) -> str:
- return get_dag_node_str(self, str(self._deployment))
- def get_deployment_name(self):
- return self._deployment.name
|
