ContextQMD
Back to Prefect

task_workers

docs/v3/api-ref/python/prefect-server-models-task_workers.mdx

3.6.30.dev33.1 KB
Original Source

prefect.server.models.task_workers

Functions

observe_worker <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L86" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
observe_worker(task_keys: List[TaskKey], worker_id: WorkerId) -> None

forget_worker <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L93" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
forget_worker(worker_id: WorkerId) -> None

get_workers_for_task_keys <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L99" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
get_workers_for_task_keys(task_keys: List[TaskKey]) -> List[TaskWorkerResponse]

get_all_workers <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L105" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
get_all_workers() -> List[TaskWorkerResponse]

Classes

TaskWorkerResponse <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L16" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

InMemoryTaskWorkerTracker <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L22" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

Methods:

forget_worker <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L39" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
forget_worker(self, worker_id: WorkerId) -> None

get_all_workers <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L60" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
get_all_workers(self) -> List[TaskWorkerResponse]

get_workers_for_task_keys <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L51" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
get_workers_for_task_keys(self, task_keys: List[TaskKey]) -> List[TaskWorkerResponse]

observe_worker <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L28" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
observe_worker(self, task_keys: List[TaskKey], worker_id: WorkerId) -> None

reset <sup><a href="https://github.com/PrefectHQ/prefect/blob/main/src/prefect/server/models/task_workers.py#L74" target="_blank"><Icon icon="github" style="width: 14px; height: 14px;" /></a></sup>

python
reset(self) -> None

Testing utility to reset the state of the task worker tracker