Skip to content

Commit 5a01c93

Browse files
committed
Fix failed test RecordingTest.test_describe
1 parent 6e8a500 commit 5a01c93

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

sdks/python/apache_beam/runners/interactive/interactive_environment.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -370,6 +370,8 @@ def set_cache_manager(self, cache_manager, pipeline):
370370
# current cache_manager is not None.
371371
self.cleanup(pipeline)
372372
self._cache_managers[str(id(pipeline))] = cache_manager
373+
# Re-track the user pipeline because the self.cleanup() call above evicts it.
374+
self.add_user_pipeline(pipeline)
373375

374376
def get_cache_manager(self, pipeline, create_if_absent=False):
375377
"""Gets the cache manager held by current Interactive Environment for the

0 commit comments

Comments
 (0)