prefect.server.models.task_workers

Functions

observe_worker

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

forget_worker

forget_worker(worker_id: WorkerId) -> None

get_workers_for_task_keys

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

get_all_workers

get_all_workers() -> List[TaskWorkerResponse]

Classes

TaskWorkerResponse

InMemoryTaskWorkerTracker

Methods:

forget_worker

forget_worker(self, worker_id: WorkerId) -> None

get_all_workers

get_all_workers(self) -> List[TaskWorkerResponse]

get_workers_for_task_keys

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

observe_worker

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

reset

reset(self) -> None
Testing utility to reset the state of the task worker tracker