Skip to content

Commit

Permalink
Add quiet parameter to KubeCluster (#840)
Browse files Browse the repository at this point in the history
* Add quiet parameter to KubeCluster

* Address feedback

---------

Co-authored-by: Jacob Tomlinson <jtomlinson@nvidia.com>
  • Loading branch information
droctothorpe and jacobtomlinson committed Jan 5, 2024
1 parent ff71342 commit eeb3c40
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion dask_kubernetes/operator/kubecluster/kubecluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,6 +115,9 @@ class KubeCluster(Cluster):
used to create the cluster instead of generating one from the other keyword arguments.
scheduler_forward_port: int (optional)
The port to use when forwarding the scheduler dashboard. Will utilize a random port by default
quiet: bool
If enabled, suppress all printed output.
Defaults to ``False``.
**kwargs: dict
Additional keyword arguments to pass to LocalCluster
Expand Down Expand Up @@ -175,6 +178,7 @@ def __init__(
jupyter: bool = False,
loop: Optional[IOLoop] = None,
asynchronous: bool = False,
quiet: bool = False,
**kwargs,
):
name = dask.config.get("kubernetes.name", override_with=name)
Expand Down Expand Up @@ -258,7 +262,9 @@ def __init__(
self._rich_spinner = Spinner("dots", speed=0.5)
self._startup_component_status: dict = {}

super().__init__(name=name, loop=loop, asynchronous=asynchronous, **kwargs)
super().__init__(
name=name, loop=loop, asynchronous=asynchronous, quiet=quiet, **kwargs
)

# If https://github.com/dask/distributed/pull/7941 is merged we can
# simplify the next 8 lines to ``if not self.called_from_running_loop:``
Expand Down

0 comments on commit eeb3c40

Please sign in to comment.