databricks.hooks.databricks

Module Contents

Classes

DatabricksHookAsync

Interact with Databricks.

Attributes

USER_AGENT_HEADER

databricks.hooks.databricks.USER_AGENT_HEADER
class databricks.hooks.databricks.DatabricksHookAsync

Bases: airflow.providers.databricks.hooks.databricks.DatabricksHook

Interact with Databricks.

Parameters

  • databricks_conn_id (str) – Reference to the Databricks connection.

  • timeout_seconds (int) – The amount of time in seconds the requests library will wait before timing-out.

  • retry_limit (int) – The number of times to retry the connection in case of service outages.

  • retry_delay (float) – The number of seconds to wait between retries (it might be a floating point number).

async get_run_state_async(self, run_id)

Retrieves run state of the run using an asynchronous api call. :param run_id: id of the run :return: state of the run