def request_headers(self):
request_headers = {}
if databricks_utils.is_in_databricks_notebook():
request_headers["notebook_id"] = databricks_utils.get_notebook_id()
if databricks_utils.is_in_databricks_job():
request_headers["job_id"] = databricks_utils.get_job_id()
request_headers["job_run_id"] = databricks_utils.get_job_run_id()
request_headers["job_type"] = databricks_utils.get_job_type()
if databricks_utils.is_in_cluster():
request_headers["cluster_id"] = databricks_utils.get_cluster_id()
return request_headers
def tags(self):
cluster_id = databricks_utils.get_cluster_id()
tags = {}
if cluster_id is not None:
tags[MLFLOW_DATABRICKS_CLUSTER_ID] = cluster_id
return tags