Files
ray-project--ray/python/ray/serve/_private/benchmarks/serialization/common.py
T
2026-07-13 13:17:40 +08:00

30 lines
802 B
Python

from dataclasses import dataclass
from typing import List, Optional
from pydantic import BaseModel
#
# NOTE: PLEASE READ CAREFULLY BEFORE CHANGING
#
# Payloads in this module are purposefully extracted from benchmark file to force
# Ray's cloudpickle behavior when it does NOT serialize the class definition itself
# along with its payload (instead relying on it being imported)
#
class PayloadPydantic(BaseModel):
text: Optional[str] = None
floats: Optional[List[float]] = None
ints: Optional[List[int]] = None
ts: Optional[float] = None
reason: Optional[str] = None
@dataclass
class PayloadDataclass:
text: Optional[str] = None
floats: Optional[List[float]] = None
ints: Optional[List[int]] = None
ts: Optional[float] = None
reason: Optional[str] = None