[BEAM-13669] Install Python wheel and dependencies to local venv in SDK harness

• Type: Improvement
• Status: Open
• Priority: P1
• Resolution: Unresolved
• Affects Version/s: 2.35.0
• Component/s: sdk-py-harness
• Labels: usability
• Assignee: Unassigned
• Reporter: Janek Bevendorff
• Created: 17/Jan/22 09:28
• Updated: 01/Sep/24 06:54

Description

With the --setup-file option, the Python SDK harness compiles and installs a wheel and its package dependencies on the taskmanager SDK sidecar containers. Unfortunately, this installation is global and persistent, so consecutive job submissions will reuse the previously installed wheel instead of reinstalling or updating it. This makes it impossible to submit updated code (beyond the main .py file, which will always be fresh) without deleting and recreating all taskmanager containers. It also messes with dependencies of other jobs and can cause inconsistencies or errors that are extremely hard to debug.

Suggestion: Create a temporary venv per job, install packages there, delete venv after job completion or failure.

Fixes: Stale or conflicting dependencies in persistent SDK containers. Pip will not reinstall existing packages by default, even if their version numbers differ and the number of installed dependency packages grows unnecessarily over time. There is also no point in keeping the --setup-file wheel installed for all eternity.

Issue Links

• relates to: BEAM-12792 Multiple jobs running on Flink session cluster reuse the persistent Python environment.