:py:mod:`astronomer.providers.databricks.hooks.databricks`
==========================================================

.. py:module:: astronomer.providers.databricks.hooks.databricks


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   astronomer.providers.databricks.hooks.databricks.DatabricksHookAsync




Attributes
~~~~~~~~~~

.. autoapisummary::

   astronomer.providers.databricks.hooks.databricks.USER_AGENT_HEADER


.. py:data:: USER_AGENT_HEADER

   

.. py:class:: DatabricksHookAsync(databricks_conn_id = BaseDatabricksHook.default_conn_name, timeout_seconds = 180, retry_limit = 3, retry_delay = 1.0, retry_args = None, caller = 'DatabricksHook')

   Bases: :py:obj:`airflow.providers.databricks.hooks.databricks.DatabricksHook`

   Interact with Databricks.

   :param databricks_conn_id: Reference to the Databricks connection.
   :type databricks_conn_id: str
   :param timeout_seconds: The amount of time in seconds the requests library
       will wait before timing-out.
   :type timeout_seconds: int
   :param retry_limit: The number of times to retry the connection in case of
       service outages.
   :type retry_limit: int
   :param retry_delay: The number of seconds to wait between retries (it
       might be a floating point number).
   :type retry_delay: float

   .. py:method:: get_run_state_async(run_id)
      :async:

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


   .. py:method:: get_run_response(run_id)
      :async:

      Makes Async API call to get the run state info.

      :param run_id: id of the run


   .. py:method:: get_run_output_response(task_run_id)
      :async:

      Retrieves run output of the run.

      :param task_run_id: id of the run