diff --git a/airflow/providers/apache/livy/hooks/livy.py b/airflow/providers/apache/livy/hooks/livy.py index b7aa65da20e22..ee453e6a1333c 100644 --- a/airflow/providers/apache/livy/hooks/livy.py +++ b/airflow/providers/apache/livy/hooks/livy.py @@ -49,6 +49,10 @@ class LivyHook(HttpHook, LoggingMixin): :param livy_conn_id: reference to a pre-defined Livy Connection. :type livy_conn_id: str + :param extra_options: A dictionary of options passed to Livy. + :type extra_options: Dict[str, Any] + :param extra_headers: A dictionary of headers passed to the HTTP request to livy. + :type extra_headers: Dict[str, Any] .. seealso:: For more details refer to the Apache Livy API reference: diff --git a/airflow/providers/apache/livy/operators/livy.py b/airflow/providers/apache/livy/operators/livy.py index d271f2cee25d0..129776900ec92 100644 --- a/airflow/providers/apache/livy/operators/livy.py +++ b/airflow/providers/apache/livy/operators/livy.py @@ -65,8 +65,11 @@ class LivyOperator(BaseOperator): :type livy_conn_id: str :param polling_interval: time in seconds between polling for job completion. Don't poll for values >=0 :type polling_interval: int - :type extra_options: A dictionary of options, where key is string and value + :param extra_options: A dictionary of options, where key is string and value depends on the option that's being modified. + :type extra_options: Dict[str, Any] + :param extra_headers: A dictionary of headers passed to the HTTP request to livy. + :type extra_headers: Dict[str, Any] """ template_fields = ('spark_params',)