retrieved_user_pipeline.visit(CacheableUnboundedPCollectionVisitor()) AttributeError: 'NoneType' object has no attribute 'visit'

Traceback (most recent call last):
File "/build/work/f1a2b3ea7c34e8e49f1a90317e5acde7889a/google3/runfiles/google3/photos/vision/features/delf/extract/global_descriptor/python/revisited_datasets/beam/eval_utils_test.py", line 614, in testProduceOutputVisualization
_ = pipeline.run()
File "/build/work/f1a2b3ea7c34e8e49f1a90317e5acde7889a/google3/runfiles/google3/third_party/py/apache_beam/pipeline.py", line 553, in run
return self.runner.run_pipeline(self, self._options)
File "/build/work/f1a2b3ea7c34e8e49f1a90317e5acde7889a/google3/runfiles/google3/third_party/py/apache_beam/runners/interactive/interactive_runner.py", line 136, in run_pipeline
inst.watch_sources(pipeline)
File "/build/work/f1a2b3ea7c34e8e49f1a90317e5acde7889a/google3/runfiles/google3/third_party/py/apache_beam/runners/interactive/pipeline_instrument.py", line 1008, in watch_sources
retrieved_user_pipeline.visit(CacheableUnboundedPCollectionVisitor())
AttributeError: 'NoneType' object has no attribute 'visit'

Probably related to this change: https://github.com/apache/beam/pull/13335