def b_op(
field_x: {
'customized_type': {
'openapi_schema_validator':
'{"type": "string", "pattern": "^gcs://.*$"}'
}
}, field_y: Integer(), field_z: GCSPath()
) -> {
'output_model_uri': 'GcsUri'
}:
return ContainerOp(name='operator b',
image='gcr.io/ml-pipeline/component-a',
command=[
'python3',
field_x,
],
arguments=[
'--field-y',
field_y,
'--field-z',
field_z,
],
file_outputs={
'output_model_uri': '/schema.txt',
})
def b_op(
field_x: {
'customized_type': {
'property_a': 'value_a',
'property_b': 'value_b'
}
}, field_y: Integer(), field_z: GCSPath()
) -> {
'output_model_uri': 'GcsUri'
}:
return ContainerOp(name='operator b',
image='gcr.io/ml-pipeline/component-a',
command=[
'python3',
field_x,
],
arguments=[
'--field-y',
field_y,
'--field-z',
field_z,
],
file_outputs={
'output_model_uri': '/schema.txt',
})
def test_class_to_dict(self):
"""Test _class_to_dict function."""
gcspath_dict = GCSPath().to_dict()
golden_dict = {
'GCSPath': {
'openapi_schema_validator': {
"type": "string",
"pattern": "^gs://.*$"
}
}
}
self.assertEqual(golden_dict, gcspath_dict)
def a_op(field_l: Integer()) -> {'field_m': GCSPath(), 'field_n': {'customized_type': {'property_a': 'value_a', 'property_b': 'value_b'}}, 'field_o': 'GcsUri'}:
return ContainerOp(
name = 'operator a',
image = 'gcr.io/ml-pipeline/component-a',
arguments = [
'--field-l', field_l,
],
file_outputs = {
'field_m': '/schema.txt',
'field_n': '/feature.txt',
'field_o': '/output.txt'
}
)
def test_class_to_dict(self):
"""Test _class_to_dict function."""
gcspath_dict = _instance_to_dict(GCSPath())
golden_dict = {'GCSPath': {}}
self.assertEqual(golden_dict, gcspath_dict)
