(env, ids: Union[int, List[int]], target_states: Union[str, List[str]], **kwargs)
| 77 | |
| 78 | |
| 79 | def wait_for_worker_state(env, ids: Union[int, List[int]], target_states: Union[str, List[str]], **kwargs): |
| 80 | wait_for_state(env, ids, target_states, ["worker", "list", "--all"], 1, **kwargs) |
| 81 | |
| 82 | |
| 83 | def wait_for_task_state(env, job_id: int, task_ids: Union[int, List[int]], states: Union[str, List[str]], **kwargs): |