Skip to content

Commit

Permalink
Add a check for string input to allow a single string
Browse files Browse the repository at this point in the history
Signed-off-by: Vibhu Jawa <[email protected]>
  • Loading branch information
VibhuJawa committed Oct 8, 2024
1 parent 494ada6 commit 94982f6
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions dask_cuda/local_cuda_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,6 +270,9 @@ def __init__(
n_workers = len(CUDA_VISIBLE_DEVICES)
if n_workers < 1:
raise ValueError("Number of workers cannot be less than 1.")

if isinstance(rmm_allocator_external_lib_list, str):
rmm_allocator_external_lib_list = []
# Set nthreads=1 when parsing mem_limit since it only depends on n_workers
logger = logging.getLogger(__name__)
self.memory_limit = parse_memory_limit(
Expand Down

0 comments on commit 94982f6

Please sign in to comment.