如何在Databricks笔记本中获取运行参数和运行ID？ [英] How do you get the run parameters and runId within Databricks notebook?

问题描述

将Databricks笔记本作为作业运行时，您可以指定可在笔记本代码中使用的作业或运行参数。然而，从文档中并不清楚您实际是如何获取它们的。我希望能够获得所有参数以及作业ID和运行ID。

推荐答案

作业/运行参数

当笔记本作为作业运行时，任何job parameters都可以使用Databricks自动提供和导入的dbutils包作为词典获取。代码如下：

run_parameters = dbutils.notebook.entry_point.getCurrentBindings()

如果作业参数为{"foo": "bar"}，则上面代码的结果为{'foo': 'bar'}。请注意，数据库仅允许将作业参数从str映射到str，因此键和值始终为字符串。

请注意，如果笔记本以交互方式运行(而不是作为作业)，则词典将为空。getCurrentBinding()方法似乎也适用于获取笔记本的任何活动widget values(交互运行时)。

获取jobID和runID

要获取jobId和runId，您可以从dbutils获取包含该信息的上下文json。(改编自databricks forum)：

import json
context_str = dbutils.notebook.entry_point.getDbutils().notebook().getContext().toJson()
context = json.loads(context_str)
run_id_obj = context.get('currentRunId', {})
run_id = run_id_obj.get('id', None) if run_id_obj else None
job_id = context.get('tags', {}).get('jobId', None)

因此在上下文对象中，runId的按键路径为currentRunId > id，jobId的按键路径为tags > jobId。

这篇关于如何在Databricks笔记本中获取运行参数和运行ID？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！