def arg_builder(spacing=spacing):
args = protobufs.SpaceRepeatSamplerArgs()
args.spacing = spacing
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = "SpaceRepeat"
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def arg_builder(rows=rows):
args = protobufs.GatherSamplerArgs()
args.rows[:] = rows
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = 'Gather'
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def arg_builder(stride=stride):
args = protobufs.StridedSamplerArgs()
args.stride = stride
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = "Strided"
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def strided(self, stride, group_size=DEFAULT_GROUP_SIZE):
args = protobufs.StridedPartitionerArgs()
args.stride = stride
args.group_size = group_size
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = 'Strided'
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def gather(self, groups):
args = protobufs.GatherSamplerArgs()
for rows in groups:
gather_group = args.groups_add()
gather_group.rows[:] = rows
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = 'Gather'
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def arg_builder(start, end, stride):
args = protobufs.StridedRangeSamplerArgs()
args.stride = stride
args.starts.append(start)
args.ends.append(end)
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = "StridedRanges"
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def strided_ranges(self, intervals, stride):
args = protobufs.StridedRangePartitionerArgs()
args.stride = stride
for start, end in intervals:
args.starts.append(start)
args.ends.append(end)
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = 'StridedRange'
sampling_args.sampling_args = args.SerializeToString()
return sampling_args
def arg_builder():
sampling_args = protobufs.SamplingArgs()
sampling_args.sampling_function = "All"
return sampling_args