apache / apache/beam

Improve caching experience on InteractiveRunner with dataframes

Open
#20,929 0 comments 0 reactions 0 assignees View on GitHub
improvement interactive P3 python runners
Dominant language
Java
Stars
8.7k
Forks
4.7k
Avg merge
1d 20h
Merged PRs (30d)
196

Description

Reusing the default label for to_pcollection when using the interactive runner results in caching errors when used with multiple pipelines:

 

 

{{Traceback (most recent call last):
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/runners/interactive/interactive_runner_test.py", line 389, in test_dataframes_with_multi_index_get_result
pd.testing.assert_series_equal(df_expected, ib.collect(deferred_df, n=10))
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/runners/interactive/utils.py", line 247, in run_within_progress_indicator
return func(*args, **kwargs)
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/runners/interactive/interactive_beam.py", line 579, in collect
recording = recording_manager.record([pcoll], max_n=n, max_duration=duration)
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/runners/interactive/recording_manager.py", line 433, in record
self._watch(pcolls)
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/runners/interactive/recording_manager.py", line 306, in _watch
for pcoll in to_pcollection(*watched_dataframes, always_return_tuple=True):
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/dataframe/convert.py", line 196, in to_pcollection
new_results = {p: extract_input(p)
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/transforms/ptransform.py", line 1086, in __ror__
return self.transform.__ror__(pvalueish, self.label)
File "/home/srohde/Workdir/beam/sdks/python/apache_beam/transforms/ptransform.py", line 587, in __ror__
raise ValueError(
ValueError: Mixing value from different pipelines not allowed.}}

Imported from Jira [BEAM-12388](https://issues.apache.org/jira/browse/BEAM-12388). Original Jira may contain additional context.
Reported by: rohdesam.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.