Update parse_bytes imports to resolve deprecation warnings (#756)

rapidsai · Jun 29, 2021 · 27f39c7 · 27f39c7
1 parent c9b4f94
commit 27f39c7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/benchmarks/send-recv-core.py b/benchmarks/send-recv-core.py
@@ -44,7 +44,7 @@
 from threading import Lock
 from time import perf_counter as clock
 
-from distributed.utils import format_bytes, parse_bytes
+from dask.utils import format_bytes, parse_bytes
 
 import ucp
 from ucp._libs import ucx_api

diff --git a/benchmarks/send-recv.py b/benchmarks/send-recv.py
@@ -44,7 +44,7 @@
 import os
 from time import perf_counter as clock
 
-from distributed.utils import format_bytes, parse_bytes
+from dask.utils import format_bytes, parse_bytes
 
 import ucp
 

diff --git a/debug-tests/debug_utils.py b/debug-tests/debug_utils.py
@@ -5,7 +5,7 @@
 import cupy
 from utils import get_num_gpus
 
-from distributed.utils import parse_bytes
+from dask.utils import parse_bytes
 
 import rmm