Skip to content

Commit 939bac6

Browse files
committed
Daskify reprojection
1 parent 90c8fd7 commit 939bac6

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

spectral_cube/dask_spectral_cube.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1495,16 +1495,11 @@ def reproject(self, header, order='bilinear', use_memmap=False,
14951495
raise ImportError("Requires the reproject package to be"
14961496
" installed.")
14971497

1498-
reproj_kwargs = kwargs
1499-
# Need version > 0.2 to work with cubes, >= 0.5 for memmap
1498+
# Need version > 0.12 to work with dask
15001499
from distutils.version import LooseVersion
1501-
if LooseVersion(version) < "0.5":
1502-
raise Warning("Requires version >=0.5 of reproject. The current "
1500+
if LooseVersion(version) < "0.12":
1501+
raise Warning("Requires version >=0.12 of reproject. The current "
15031502
"version is: {}".format(version))
1504-
elif LooseVersion(version) >= "0.6":
1505-
pass # no additional kwargs, no warning either
1506-
else:
1507-
reproj_kwargs['independent_celestial_slices'] = True
15081503

15091504
from reproject import reproject_interp
15101505

@@ -1515,10 +1510,8 @@ def reproject(self, header, order='bilinear', use_memmap=False,
15151510
shape_out = tuple([header['NAXIS{0}'.format(i + 1)] for i in
15161511
range(header['NAXIS'])][::-1])
15171512

1518-
print(self._get_filled_data())
1519-
15201513
newcube, newcube_valid = reproject_interp((self._get_filled_data() if filled else self._data, self.header),
1521-
newwcs, shape_out=shape_out, block_size=(256, 256, 256), **kwargs
1514+
newwcs, shape_out=shape_out, block_size=(256, 256, 256), return_type='dask',
15221515
)
15231516

15241517
return self._new_cube_with(data=newcube,
@@ -1529,7 +1522,6 @@ def reproject(self, header, order='bilinear', use_memmap=False,
15291522
)
15301523

15311524

1532-
15331525
class DaskVaryingResolutionSpectralCube(DaskSpectralCubeMixin, VaryingResolutionSpectralCube):
15341526

15351527
def __init__(self, data, *args, **kwargs):

0 commit comments

Comments
 (0)