30 lines
802 B
Python
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
|