distributor.task_instance_batch

Attributes

logger

Classes

TaskInstanceBatch

Initialization of the TaskInstanceBatch object.

Module Contents

distributor.task_instance_batch.logger

class distributor.task_instance_batch.TaskInstanceBatch(array_id: int, array_name: str, array_batch_num: int, task_resources_id: int, requester: jobmon.core.requester.Requester)

Initialization of the TaskInstanceBatch object.

array_id

array_name

batch_number

task_resources_id

task_instances: Set[jobmon.distributor.distributor_task_instance.DistributorTaskInstance]

requester

property submission_name: str

property requested_resources: Dict

add_task_instance(task_instsance: jobmon.distributor.distributor_task_instance.DistributorTaskInstance) → None

load_requested_resources() → None

prepare_task_instance_batch_for_launch() → None: Add the current batch number to the current set of registered task instance ids.

set_distributor_ids(distributor_id_map: Dict) → None

Set the distributor_ids on the task instances in the array.

Parameters:: distributor_id_map – map of array_step_id to distributor_id

transition_to_launched(next_report_by: float) → None: Transition all associated task instances to LAUNCHED state.

log_distributor_ids() → None: Log the distributor ID in the database for all task instances in the batch.

transition_to_killed() → None: Mark all TIs in this batch as killed in the DB (ERROR_FATAL).