mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
31 lines
704 B
Python
31 lines
704 B
Python
|
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
|