class Barrier: """ barrier """ def __init__(self, id): self.id = id def __str__(self): return "Barrier [id:%s]" % self.id class OpCheckpointInfo: """ operator checkpoint info """ def __init__(self, operator_point=None, input_points=None, output_points=None, checkpoint_id=None): if input_points is None: input_points = {} if output_points is None: output_points = {} self.operator_point = operator_point self.input_points = input_points self.output_points = output_points self.checkpoint_id = checkpoint_id