pep8 modification
import sys
import dask
+import dask.array as da
from dask.diagnostics import ProgressBar
import warnings
import numpy as np
@@ -1838,10 +1839,10 @@ def snow_partitioning_parallel(im,
tup.dt = dt
# --------------------------------------------------------------------------
# Applying snow to image chunks
- im = dask.array.from_array(dt, chunks=chunk_shape)
- im = dask.array.overlap.overlap(im, depth=depth, boundary='none')
+ im = da.from_array(dt, chunks=chunk_shape)
+ im = da.overlap.overlap(im, depth=depth, boundary='none')
im = im.map_blocks(chunked_snow, r_max=r_max, sigma=sigma)
- im = dask.array.overlap.trim_internal(im, trim_depth, boundary='none')
+ im = da.overlap.trim_internal(im, trim_depth, boundary='none')
if mode == 'serial':
num_workers = 1
elif mode == 'parallel':
