flytekitplugins.dask.Scheduler#

class flytekitplugins.dask.Scheduler(image=None, requests=None, limits=None)[source]#

Configuration for the scheduler pod

Parameters
  • image (Optional[str]) – Custom image to use. If None, will use the same image the task was registered with. Optional, defaults to None. The image must have dask[distributed] installed and should have the same Python environment as the rest of the cluster (job runner pod + worker pods).

  • requests (Optional[flytekit.core.resources.Resources]) – Resources to request for the scheduler pod. If None, the requests passed into the task will be used. Optional, defaults to None.

  • limits (Optional[flytekit.core.resources.Resources]) – Resource limits for the scheduler pod. If None, the limits passed into the task will be used. Optional, defaults to None.

Return type

None

Methods

Attributes

image: Optional[str] = None
limits: Optional[flytekit.core.resources.Resources] = None
requests: Optional[flytekit.core.resources.Resources] = None