AnandInguva commented on code in PR #30085:
URL: https://github.com/apache/beam/pull/30085#discussion_r1465049723
##########
sdks/python/apache_beam/ml/transforms/base_test.py:
##########
@@ -443,6 +442,26 @@ def test_handler_on_multiple_columns(self):
equal_to(expected_data),
)
+ def test_handler_on_columns_not_exist_in_input_data(self):
+ data = [
+ {
+ 'x': "Hello world", 'y': "Apache Beam"
Review Comment:
I made changes according to
1. Add the first record to the dictionary
2. For each subsequent record, before adding it to the dictionary check that
it contains all the same fields and doesn't contain extra fields. Throw if it
doesn't satisfy these conditions, else add it to the dictionary.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
