Fix FutureWarning format_bytes/parse_bytes (#506)

dask · Jul 13, 2021 · 0b0bdfa · 0b0bdfa
1 parent 1f8edf0
commit 0b0bdfa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dask_jobqueue/core.py b/dask_jobqueue/core.py
@@ -11,11 +11,13 @@
 
 import dask
 
+from dask.utils import format_bytes, parse_bytes
+
 from distributed.core import Status
 from distributed.deploy.spec import ProcessInterface, SpecCluster
 from distributed.deploy.local import nprocesses_nthreads
 from distributed.scheduler import Scheduler
-from distributed.utils import format_bytes, parse_bytes, tmpfile
+from distributed.utils import tmpfile
 
 logger = logging.getLogger(__name__)
 

diff --git a/dask_jobqueue/htcondor.py b/dask_jobqueue/htcondor.py
@@ -3,7 +3,7 @@
 import shlex
 
 import dask
-from distributed.utils import parse_bytes
+from dask.utils import parse_bytes
 
 from .core import JobQueueCluster, Job, job_parameters, cluster_parameters