https://flyte.org logo
#ray-integration
Title
# ray-integration
f

Franco Bocci

09/13/2023, 1:41 PM
Team! One doubt over here. Is there a way to define resources for the ray workers being created? If I got it right, we can define the amount of replicas, but I did not get how to set the resources for those workers
Copy code
@dataclass
class WorkerNodeConfig:
    group_name: str
    replicas: int
    min_replicas: typing.Optional[int] = None
    max_replicas: typing.Optional[int] = None
    ray_start_params: typing.Optional[typing.Dict[str, str]] = None
y

Yicheng Lu

09/13/2023, 1:49 PM
one way is setting pod limits.
f

Franco Bocci

09/13/2023, 1:50 PM
Would this apply to the head or to the workers?
y

Yicheng Lu

09/13/2023, 1:54 PM
Will apply to both. Currently, there is no easy way to set head and workers resources differently.
f

Franco Bocci

09/13/2023, 2:18 PM
Okay, makes sense. Appreciate it 🙂
3 Views