diff --git a/examples/ucx/client_initialize.py b/examples/ucx/client_initialize.py index 2a7c6c053..0fe28f5fa 100644 --- a/examples/ucx/client_initialize.py +++ b/examples/ucx/client_initialize.py @@ -44,12 +44,12 @@ def main( # initialize client client = Client(address) - # client code here + # user code here rs = da.random.RandomState(RandomState=cupy.random.RandomState) x = rs.random((10000, 10000), chunks=1000) x.sum().compute() - # shutdown client + # shutdown cluster client.shutdown() diff --git a/examples/ucx/local_cuda_cluster.py b/examples/ucx/local_cuda_cluster.py index ad969857d..bf91c6e62 100644 --- a/examples/ucx/local_cuda_cluster.py +++ b/examples/ucx/local_cuda_cluster.py @@ -64,12 +64,12 @@ def main( # initialize client client = Client(cluster) - # client code here + # user code here rs = da.random.RandomState(RandomState=cupy.random.RandomState) x = rs.random((10000, 10000), chunks=1000) x.sum().compute() - # shutdown client + # shutdown cluster client.shutdown()