Fix cli problem

Signed-off-by: Vibhu Jawa <[email protected]>
rapidsai · Oct 8, 2024 · d046478 · d046478
1 parent a2bf28f
commit d046478
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dask_cuda/cli.py b/dask_cuda/cli.py
@@ -166,11 +166,13 @@ def cuda():
 )
 @click.option(
     "--set-rmm-allocator-for-libs",
+    "rmm_allocator_external_lib_list",
     default=None,
     show_default=True,
     help="""
     Set RMM as the allocator for external libraries. Provide a comma-separated
-    list of libraries to set, e.g., "torch,cupy". Supported options are: torch, cupy.""",
+    list of libraries to set, e.g., "torch,cupy".
+    Supported options are: torch, cupy.""",
 )
 @click.option(
     "--rmm-release-threshold",