Skip to content

Commit 5872b38

Browse files
authored
Init RuntimeContext from debug notebook to simplify interactive debugging flows (#2253)
UCX manages all object dependencies via subclasses of `GlobalContext`. All UCX workflows use `RuntimeContext` instance for any object lookup, so this PR simplifies debugging by pre-initializing `RuntimeContext` the correct way.
1 parent 289d03b commit 5872b38

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/databricks/labs/ucx/installer/workflows.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,17 +78,19 @@
7878
7979
import logging
8080
from pathlib import Path
81+
from databricks.sdk.config import with_user_agent_extra
8182
from databricks.labs.blueprint.installation import Installation
8283
from databricks.labs.blueprint.logger import install_logger
8384
from databricks.labs.ucx.__about__ import __version__
84-
from databricks.labs.ucx.config import WorkspaceConfig
85-
from databricks.sdk import WorkspaceClient
85+
from databricks.labs.ucx.contexts.workflow_task import RuntimeContext
8686
8787
install_logger()
88+
with_user_agent_extra("cmd", "debug-notebook")
8889
logging.getLogger("databricks").setLevel("DEBUG")
8990
90-
cfg = Installation.load_local(WorkspaceConfig, Path("/Workspace{config_file}"))
91-
ws = WorkspaceClient()
91+
# ctx.<TAB> to see all available objects for you to
92+
named_parameters = dict(config="/Workspace{config_file}")
93+
ctx = RuntimeContext(named_parameters)
9294
9395
print(__version__)
9496
"""

0 commit comments

Comments
 (0)