[Serve] Alias ray.serve.dag.InputNode (#24630)

doc/source/serve/deployment-graph.md

@@ -17,4 +17,4 @@ kernelspec:
 # Deployment Graph
 
 - [E2E Tutorials](./deployment-graph/deployment-graph-e2e-tutorial.md)
-- [User Guides](./deployment-graph/deployment-graph-user-guides.md)
+- [User Guides](./deployment-graph/deployment-graph-user-guides.md)

doc/source/serve/deployment-graph/deployment-graph-e2e-tutorial.md

@@ -88,7 +88,7 @@ import asyncio
 from ray import serve
 # We will later move Ray DAG related components
 # out of experimental in later stable release
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 
 @serve.deployment
 async def preprocessor(input_data: str):
@@ -429,7 +429,7 @@ import starlette
 
 import ray
 from ray import serve
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray.serve.drivers import DAGDriver
 from ray.serve.http_adapters import json_request
 

doc/source/serve/http-servehandle.md

@@ -236,7 +236,7 @@ For example, the json request adapters parse JSON in HTTP body:
 ```python
 from ray.serve.drivers import DAGDriver
 from ray.serve.http_adapters import json_request
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 
 with InputNode() as input_node:
     ...

python/ray/serve/dag.py

@@ -0,0 +1,5 @@
+from ray.experimental.dag.input_node import InputNode
+
+__all__ = [
+    "InputNode",
+]

python/ray/serve/pipeline/tests/test_deployment_node.py

@@ -2,7 +2,7 @@ import pytest
 
 import ray
 from ray import serve
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray.serve.pipeline.deployment_node import (
     DeploymentNode,
 )

python/ray/serve/tests/test_model_wrappers.py

@@ -11,7 +11,7 @@ from ray.ml.checkpoint import Checkpoint
 from ray.ml.predictor import DataBatchType, Predictor
 from ray.serve.model_wrappers import ModelWrapperDeployment
 from ray.serve.pipeline.api import build
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray.serve.deployment_graph import RayServeDAGHandle
 from ray.serve.http_adapters import json_to_ndarray
 import ray

python/ray/serve/tests/test_pipeline_deployment_config.py

@@ -5,7 +5,7 @@ from typing import Any
 
 from ray import serve
 from ray.serve.deployment_graph import RayServeDAGHandle
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray.serve.pipeline.api import build as pipeline_build
 
 

python/ray/serve/tests/test_pipeline_driver.py

@@ -11,7 +11,7 @@ from starlette.testclient import TestClient
 
 from ray.serve.drivers import DAGDriver, SimpleSchemaIngress, load_http_adapter
 from ray.serve.http_adapters import json_request
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray import serve
 import ray
 from ray._private.test_utils import wait_for_condition

python/ray/serve/tests/test_pipeline_handle_serde.py

@@ -1,7 +1,7 @@
 import sys
 
 import pytest
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray.serve.pipeline.api import build as pipeline_build
 
 import ray

python/ray/serve/tests/test_pipeline_ingress_deployment.py

@@ -3,7 +3,7 @@ import sys
 
 from ray.serve.pipeline.api import get_and_validate_ingress_deployment
 from ray.serve.pipeline.api import build as pipeline_build
-from ray.experimental.dag.input_node import InputNode
+from ray.serve.dag import InputNode
 from ray import serve