Skip to content

Commit d37c808

Browse files
committed
fix interactive enviornemnt circular import
1 parent 96d195b commit d37c808

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838
from apache_beam.runners.direct import direct_runner
3939
from apache_beam.runners.interactive import cache_manager as cache
4040
from apache_beam.runners.interactive.messaging.interactive_environment_inspector import InteractiveEnvironmentInspector
41-
from apache_beam.runners.interactive.recording_manager import RecordingManager
4241
from apache_beam.runners.interactive.sql.sql_chain import SqlChain
4342
from apache_beam.runners.interactive.user_pipeline_tracker import UserPipelineTracker
4443
from apache_beam.runners.interactive.utils import assert_bucket_exists
@@ -428,6 +427,10 @@ def set_recording_manager(self, recording_manager, pipeline):
428427

429428
def get_recording_manager(self, pipeline, create_if_absent=False):
430429
"""Gets the recording manager for the given pipeline."""
430+
# Allow initial module loading to be complete and not have a circular
431+
# import.
432+
from apache_beam.runners.interactive.recording_manager import RecordingManager
433+
431434
recording_manager = self._recording_managers.get(str(id(pipeline)), None)
432435
if not recording_manager and create_if_absent:
433436
# Get the pipeline variable name for the user. This is useful if the user

0 commit comments

Comments
 (0)