From 856c4fe60855906f8eddaab286adca0126104cd3 Mon Sep 17 00:00:00 2001 From: Hugo MacDermott-Opeskin Date: Fri, 26 May 2023 22:04:18 +1000 Subject: [PATCH] Add `__main__` entrypoint to dask-cuda-worker CLI (#1181) Fixes #1180 Making the CLI runnable with `python -m ` so that we can use the same call for both CLIs in `dask-jobqueue` Authors: - Hugo MacDermott-Opeskin (https://github.com/hmacdope) - Peter Andreas Entschev (https://github.com/pentschev) Approvers: - Lawrence Mitchell (https://github.com/wence-) - Peter Andreas Entschev (https://github.com/pentschev) URL: https://github.com/rapidsai/dask-cuda/pull/1181 --- dask_cuda/cli.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/dask_cuda/cli.py b/dask_cuda/cli.py index 128da2078..5ab74e1f0 100644 --- a/dask_cuda/cli.py +++ b/dask_cuda/cli.py @@ -499,3 +499,7 @@ def config( else: client = Client(scheduler, security=security) print_cluster_config(client) + + +if __name__ == "__main__": + worker()