From 40c8b1a35ddbe49b65757e5c57a67516ba2d8184 Mon Sep 17 00:00:00 2001 From: Philippe Gagnon <12717218+pgagnon@users.noreply.github.com> Date: Sat, 19 Jun 2021 10:29:58 -0400 Subject: [PATCH] Add undocumented parameters to ``LivyOperator`` and ``LivyHook`` docstrings --- airflow/providers/apache/livy/hooks/livy.py | 4 ++++ airflow/providers/apache/livy/operators/livy.py | 5 ++++- 2 files changed, 8 insertions(+), 1 deletion(-) 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',)