From 85f929a0e66458a001219ee2731e9033711ae8f8 Mon Sep 17 00:00:00 2001 From: Hussein Awala Date: Mon, 26 Feb 2024 13:48:23 +0100 Subject: [PATCH] doc: add missing parameters to docstring (#4) --- spark_on_k8s/utils/app_manager.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/spark_on_k8s/utils/app_manager.py b/spark_on_k8s/utils/app_manager.py index 2fb44fe..8437fb7 100644 --- a/spark_on_k8s/utils/app_manager.py +++ b/spark_on_k8s/utils/app_manager.py @@ -233,6 +233,7 @@ def kill_app( namespace (str): Namespace. pod_name (str): Pod name. app_id (str): App ID. + should_print (bool, optional): Whether to print logs instead of logging them. """ if pod_name is None and app_id is None: raise ValueError("Either pod_name or app_id must be specified") @@ -455,6 +456,7 @@ def spark_app_labels( Args: app_name: Name of the Spark application app_id: ID of the Spark application + extra_labels: Dictionary of extra labels to add to the labels Returns: Dictionary of labels for the Spark application resources @@ -482,6 +484,7 @@ def create_headless_service_object( app_id: ID of the Spark application namespace: Kubernetes namespace to use, defaults to "default" pod_owner_uid: UID of the pod to use as owner reference for the service + extra_labels: Dictionary of extra labels to add to the service Returns: The created headless service for the Spark application