def get_values(self, **kwargs):
self.enumerated = True
cost = self.cond_stream.function(*unwrap_inputs(self.inputs))
assert isinstance(cost, numbers.Number) and (0 <= cost)
cost = int(self.cond_stream.scale * cost)
return map(lambda f: Initialize(f, cost),
self.cond_stream.instantiate_effects(self.inputs, []))
def __init__(self, cond_stream, inputs):
Stream.__init__(self, cond_stream, inputs)
self.generator = self.cond_stream.generator_fn(
*unwrap_inputs(self.inputs))
def function(self, inputs):
output_list = self.cond_stream.function(*unwrap_inputs(inputs))
assert type(output_list) in (list, tuple)
return [self.wrap_outputs(outputs) for outputs in output_list]
def function(self, inputs):
outputs = self.cond_stream.function(*unwrap_inputs(inputs))
if outputs is None:
return []
return [self.wrap_outputs(outputs)]
def test(self, inputs):
truth = self.cond_stream.test(*unwrap_inputs(inputs))
if truth not in (True, False):
raise ValueError(
'Expected boolean test output but received %s' % truth)
return truth
def get_generator(self, inputs):
for output_list in self.cond_stream.generator(
*unwrap_inputs(inputs)):
assert type(output_list) in (list, tuple)
yield [self.wrap_outputs(outputs) for outputs in output_list]
def get_generator(self, inputs):
for outputs in self.cond_stream.generator(*unwrap_inputs(inputs)):
yield [] if outputs is None else [self.wrap_outputs(outputs)]