chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
# __serve_example_begin__
|
||||
import time
|
||||
from ray import serve
|
||||
|
||||
|
||||
@serve.deployment
|
||||
class Preprocessor:
|
||||
def __call__(self, input_data: str) -> str:
|
||||
# Simulate preprocessing work
|
||||
time.sleep(0.05)
|
||||
return f"preprocessed_{input_data}"
|
||||
|
||||
|
||||
@serve.deployment
|
||||
class Model:
|
||||
def __call__(self, preprocessed_data: str) -> str:
|
||||
# Simulate model inference (takes longer than preprocessing)
|
||||
time.sleep(0.1)
|
||||
return f"result_{preprocessed_data}"
|
||||
|
||||
|
||||
@serve.deployment
|
||||
class Driver:
|
||||
def __init__(self, preprocessor, model):
|
||||
self._preprocessor = preprocessor
|
||||
self._model = model
|
||||
|
||||
async def __call__(self, input_data: str) -> str:
|
||||
# Coordinate preprocessing and model inference
|
||||
preprocessed = await self._preprocessor.remote(input_data)
|
||||
result = await self._model.remote(preprocessed)
|
||||
return result
|
||||
|
||||
|
||||
app = Driver.bind(Preprocessor.bind(), Model.bind())
|
||||
# __serve_example_end__
|
||||
Reference in New Issue
Block a user