def func_m(
a: int,
b: int,
num_iterations: Parameter[int] = 'abc'
) -> OutputDict(c=float):
del num_iterations
return {'c': float(a + b)}
def func_k(
a: int,
b: int,
model: OutputArtifact[standard_artifacts.Model] = None
) -> OutputDict(c=float):
del model
return {'c': float(a + b)}
def func_j(
a: int,
b: int,
examples: InputArtifact[standard_artifacts.Examples] = 123
) -> OutputDict(c=float):
del examples
return {'c': float(a + b)}
def func_a(
examples: InputArtifact[standard_artifacts.Examples],
model: OutputArtifact[standard_artifacts.Model],
schema: InputArtifact[standard_artifacts.Schema],
statistics: OutputArtifact[standard_artifacts.ExampleStatistics],
num_steps: int
) -> OutputDict(
precision=float, recall=float, message=Text, serialized_value=bytes):
del examples, model, schema, statistics, num_steps
return {
'precision': 0.9,
'recall': 0.8,
'message': 'foo',
'serialized_value': b'bar'
}
def _injector_1(
foo: Parameter[int], bar: Parameter[Text]) -> OutputDict(
a=int, b=int, c=Text, d=bytes):
assert foo == 9
assert bar == 'secret'
return {'a': 10, 'b': 22, 'c': 'unicode', 'd': b'bytes'}
def func_a(a: int, b: int, unused_c: Text, unused_d: bytes,
unused_e: Parameter[float]) -> OutputDict(c=float):
return {'c': float(a + b)}
def func_l(a: int,
b: int,
num_iterations: int = 'abc') -> OutputDict(c=float):
del num_iterations
return {'c': float(a + b)}
def func_g(a: int, b: int) -> OutputDict(c='whatever'):
return {'c': float(a + b)}
def func_h(a: int,
b: int) -> OutputDict(c=standard_artifacts.Examples):
return {'c': float(a + b)}
def func_e(
a: int, unused_b: standard_artifacts.Examples
) -> OutputDict(c=float):
return {'c': float(a)}
def func_f(a: int, b: Dict[int, int]) -> OutputDict(c=float):
return {'c': float(a + b)}
def func_d(a: int, b) -> OutputDict(c=float):
return {'c': float(a + b)}
def func_c(a: int, b: int, **unused_kwargs) -> OutputDict(c=float):
return {'c': float(a + b)}
def func_a(a: int, b: int, unused_c: Text,
unused_d: bytes) -> OutputDict(c=float):
return {'c': float(a + b)}
def _simple_component(a: int, b: int, c: Text, d: bytes) -> OutputDict(
e=float, f=float):
del c, d
return {'e': float(a + b), 'f': float(a * b)}
def func_i(
a: int, b: int
) -> OutputDict(c=OutputArtifact[standard_artifacts.Examples]):
return {'c': float(a + b)}
def _injector_2(
examples: OutputArtifact[standard_artifacts.Examples]
) -> OutputDict(
a=int, b=float, c=Text, d=bytes, e=Text):
del examples
return {'a': 1, 'b': 2.0, 'c': '3', 'd': b'4', 'e': 'passed'}
def _injector() -> OutputDict(a=int, b=int, c=Text, d=bytes):
return {'a': 10, 'b': 22, 'c': 'unicode', 'd': b'bytes'}