Fixes yapf
diff --git a/sdks/python/apache_beam/transforms/external.py b/sdks/python/apache_beam/transforms/external.py
index d1e8ab7..4a16c22 100644
--- a/sdks/python/apache_beam/transforms/external.py
+++ b/sdks/python/apache_beam/transforms/external.py
@@ -152,7 +152,6 @@
Builds a payload for directly instantiating a Java transform using a
constructor and builder methods.
"""
-
def __init__(self, class_name):
"""
:param class_name: fully qualified name of the transform class.
@@ -171,8 +170,11 @@
if not key:
raise ValueError('Parameter name cannot be empty')
if value is None:
- raise ValueError('Received value None for key %s. None values are currently not supported' % key)
- named_fields.append((key, convert_to_typing_type(instance_to_type(value))))
+ raise ValueError(
+ 'Received value None for key %s. None values are currently not supported'
+ % key)
+ named_fields.append(
+ (key, convert_to_typing_type(instance_to_type(value))))
schema_proto = named_fields_to_schema(named_fields)
row = named_tuple_from_schema(schema_proto)(**kwargs)
@@ -208,7 +210,8 @@
:param kwargs: parameter names and values of the constructor.
"""
if self._constructor_method or self._constructor_params:
- raise ValueError('Constructor or constructor method can only be specified once')
+ raise ValueError(
+ 'Constructor or constructor method can only be specified once')
self._constructor_params = kwargs
@@ -220,12 +223,13 @@
:param kwargs: parameter names and values of the constructor method.
"""
if self._constructor_method or self._constructor_params:
- raise ValueError('Constructor or constructor method can only be specified once')
+ raise ValueError(
+ 'Constructor or constructor method can only be specified once')
self._constructor_method = method_name
self._constructor_params = kwargs
- def add_builder_method(self, method_name, **kwargs):
+ def add_builder_method(self, method_name, **kwargs):
"""
Specifies a Java builder method to be invoked after instantiating the Java
transform class. Specified builder method will be applied in order.
diff --git a/sdks/python/apache_beam/transforms/external_test.py b/sdks/python/apache_beam/transforms/external_test.py
index 69da239..889c782 100644
--- a/sdks/python/apache_beam/transforms/external_test.py
+++ b/sdks/python/apache_beam/transforms/external_test.py
@@ -409,13 +409,13 @@
return get_payload(DataclassTransform(**values))
-class JavaClassLookupPayloadBuilderTest(unittest.TestCase):
+class JavaClassLookupPayloadBuilderTest(unittest.TestCase):
def _verify_row(self, schema, row_payload, expected_values):
row = RowCoder(schema).decode(row_payload)
for attr_name, expected_value in expected_values.items():
- self.assertTrue(hasattr(row,attr_name))
+ self.assertTrue(hasattr(row, attr_name))
value = getattr(row, attr_name)
self.assertEqual(expected_value, value)
@@ -424,38 +424,47 @@
payload_builder.add_constructor(str_field='abc', int_field=123)
payload_bytes = payload_builder.payload()
- payload_from_bytes = proto_utils.parse_Bytes(payload_bytes, JavaClassLookupPayload)
+ payload_from_bytes = proto_utils.parse_Bytes(
+ payload_bytes, JavaClassLookupPayload)
self.assertTrue(isinstance(payload_from_bytes, JavaClassLookupPayload))
self._verify_row(
payload_from_bytes.constructor_schema,
- payload_from_bytes.constructor_payload,
- {'str_field': 'abc', 'int_field': 123})
+ payload_from_bytes.constructor_payload, {
+ 'str_field': 'abc', 'int_field': 123
+ })
def test_build_payload_with_constructor_method(self):
payload_builder = JavaClassLookupPayloadBuilder('dummy_class_name')
- payload_builder.add_constructor_method('dummy_constructor_method', str_field='abc', int_field=123)
+ payload_builder.add_constructor_method(
+ 'dummy_constructor_method', str_field='abc', int_field=123)
payload_bytes = payload_builder.payload()
- payload_from_bytes = proto_utils.parse_Bytes(payload_bytes, JavaClassLookupPayload)
+ payload_from_bytes = proto_utils.parse_Bytes(
+ payload_bytes, JavaClassLookupPayload)
self.assertTrue(isinstance(payload_from_bytes, JavaClassLookupPayload))
- self.assertEqual('dummy_constructor_method',
- payload_from_bytes.constructor_method)
+ self.assertEqual(
+ 'dummy_constructor_method', payload_from_bytes.constructor_method)
self._verify_row(
payload_from_bytes.constructor_schema,
- payload_from_bytes.constructor_payload,
- {'str_field': 'abc', 'int_field': 123})
+ payload_from_bytes.constructor_payload, {
+ 'str_field': 'abc', 'int_field': 123
+ })
def test_build_payload_with_builder_methods(self):
payload_builder = JavaClassLookupPayloadBuilder('dummy_class_name')
payload_builder.add_constructor(str_field='abc', int_field=123)
- payload_builder.add_builder_method('builder_method1', str_field1='abc1', int_field1=1234)
- payload_builder.add_builder_method('builder_method2', str_field2='abc2', int_field2=5678)
+ payload_builder.add_builder_method(
+ 'builder_method1', str_field1='abc1', int_field1=1234)
+ payload_builder.add_builder_method(
+ 'builder_method2', str_field2='abc2', int_field2=5678)
payload_bytes = payload_builder.payload()
- payload_from_bytes = proto_utils.parse_Bytes(payload_bytes, JavaClassLookupPayload)
+ payload_from_bytes = proto_utils.parse_Bytes(
+ payload_bytes, JavaClassLookupPayload)
self.assertTrue(isinstance(payload_from_bytes, JavaClassLookupPayload))
self._verify_row(
payload_from_bytes.constructor_schema,
- payload_from_bytes.constructor_payload,
- {'str_field': 'abc', 'int_field': 123})
+ payload_from_bytes.constructor_payload, {
+ 'str_field': 'abc', 'int_field': 123
+ })
self.assertEqual(2, len(payload_from_bytes.builder_methods))
builder_method = payload_from_bytes.builder_methods[0]
self.assertTrue(isinstance(builder_method, BuilderMethod))
@@ -463,16 +472,18 @@
self._verify_row(
builder_method.schema,
- builder_method.payload,
- {'str_field1': 'abc1', 'int_field1': 1234})
+ builder_method.payload, {
+ 'str_field1': 'abc1', 'int_field1': 1234
+ })
builder_method = payload_from_bytes.builder_methods[1]
self.assertTrue(isinstance(builder_method, BuilderMethod))
self.assertEqual('builder_method2', builder_method.name)
self._verify_row(
builder_method.schema,
- builder_method.payload,
- {'str_field2': 'abc2', 'int_field2': 5678})
+ builder_method.payload, {
+ 'str_field2': 'abc2', 'int_field2': 5678
+ })
if __name__ == '__main__':