|
15 | 15 | # limitations under the License. |
16 | 16 | # |
17 | 17 |
|
18 | | -import asyncio |
19 | 18 | import collections |
20 | 19 | from concurrent.futures import Future, ThreadPoolExecutor |
21 | 20 | import logging |
|
37 | 36 | import apache_beam as beam |
38 | 37 | from apache_beam.dataframe.frame_base import DeferredBase |
39 | 38 | from apache_beam.options import pipeline_options |
40 | | -from apache_beam.pipeline import PipelineVisitor |
41 | 39 | from apache_beam.portability.api import beam_runner_api_pb2 |
42 | 40 | from apache_beam.runners import runner |
43 | 41 | from apache_beam.runners.interactive import background_caching_job as bcj |
|
47 | 45 | from apache_beam.runners.interactive import utils |
48 | 46 | from apache_beam.runners.interactive.caching.cacheable import CacheKey |
49 | 47 | from apache_beam.runners.interactive.display.pipeline_graph import PipelineGraph |
50 | | -from apache_beam.runners.interactive.messaging.interactive_environment_inspector import meta |
51 | 48 | from apache_beam.runners.interactive.options import capture_control |
52 | 49 | from apache_beam.runners.runner import PipelineState |
53 | 50 |
|
@@ -721,10 +718,6 @@ def _get_pcoll_id_map(self): |
721 | 718 | else: |
722 | 719 | # Fallback for proto-based PipelineGraph, though less likely in this context |
723 | 720 | proto = self._pipeline_graph._pipeline_proto |
724 | | - for pid, pcoll_proto in proto.components.pcollections.items(): |
725 | | - # This is tricky, we don't have the direct Python object map here. |
726 | | - # This path is less likely to be hit if PipelineGraph is init with a beam.Pipeline |
727 | | - pass |
728 | 721 | return {v: k for k, v in pcoll_to_id.items()} |
729 | 722 |
|
730 | 723 | def _get_all_dependencies( |
|
0 commit comments