...
- Input PCollection that is fed into the cross-language transform in Python side uses a Python specific type. In this case, the PTranform that produced the input PCollection has to be updated to produce outputs that use Beam's Standard Coders to make sure Java side can interpret such coders.
- Input PCollection uses standard coders but Python type inferencing results in picking up the PickleCoder. This can usually be resolved by annotating the predecessor Python transform with the correct type annotation using the with_output_types tag. See this Jira for an example.
"Unknown Coder URN beam:coder:pickled_python:v1" when running a Java pipeline that uses Python cross-language transforms
This usually means that Python SDK was not able to properly determine a portable output PCollection type when expanding the cross-language transform. So it ended up picking the default PickleCoder.
But Java SDK is unable to interpret this, so it will fail when trying to parse the expansion response from the Python SDK.
The solution is to provide a hint to the Python SDK regarding the element type(s) of the output PCollection(s) of the cross-language transform. This can be provided using the withOutputCoder or withOutputCoders methods of the PythonExternalTransform API.
How to set Java log level from a Python pipeline that uses Java transforms
For supported runners (e.g. portable runners and Dataflow runner), you can set the log level of Java transforms in the same way of setting python module log level overrides, specifically, using the --sdk_harness_log_level_overrides pipeline option. The python_underline_style option names will be automatically translated to Java smallCamel style and recognized by the Java SDK harness.
If the runner does not support the automatic mapping of options, One can try adding the corresponding pipeline option as a local pipeline option explicitly in Python side. For example, to suppress all logs from Java package org.apache.kafka package you can do following.
- Add a Python PipelineOption that represents the corresponding Java PipelineOption available here. This can be simply added to your Python program that starts up the Beam job.
Code Block language py class JavaLoggingOptions(PipelineOptions): @classmethod def _add_argparse_args(cls, parser): parser.add_argument( '--sdkHarnessLogLevelOverrides', default={}, type=json.loads, help=( 'Java log level overrides'))
Specify the additional PipelineOption as a parameter when running the Beam pipeline.
Code Block language bash --sdkHarnessLogLevelOverrides
...
'{
...
"org.apache.kafka
...
":
...
"ERROR
...
"}
...
'
Debugging a Python Test that calls a Java transform
...