[BEAM-13625] pipeline_fragment incorrectly prunes producer transform

• Type: Bug
• Status: Resolved
• Priority: P2
• Resolution: Duplicate
• Fix Version/s: Missing
• Component/s: sdk-py-core
• Assignee: Brian Hulette
• Reporter: Brian Hulette
• Created: 10/Jan/22 17:32
• Updated: 13/Jan/22 21:49
• Resolved: 10/Jan/22 17:33

Description

Unfortunately I haven't been able to diagnose the exact issue here or come up with a minimal repro. I just have some code to reproduce in https://github.com/apache/beam/pull/16445.

That PR adds support for value_count(bins) in the DataFrame API, which for some reason is interacting poorly with pipeline pruning in interactive Beam (rehydrating the pipeline raises an error about a PCollection's producer missing). The PR also adds a test to transform_test.py that replicate the issue, as well as a temporary mitigation in pipeline_fragment.py. I think the mitigation is effectively disabling pipeline pruning, so it likely shouldn't be merged.

Issue Links

• relates to: BEAM-13624 pipeline_fragment incorrectly prunes producer transform