Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions airflow/providers/apache/livy/hooks/livy.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
5 changes: 4 additions & 1 deletion airflow/providers/apache/livy/operators/livy.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',)
Expand Down