Skip to content

Commit f440ffb

Browse files
use dtype information to setup WkwDatasetInfo
1 parent a9efb0d commit f440ffb

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

wkcuber/downsampling.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,6 @@ def downsample_cube_job(
186186
with open_wkw(source_wkw_info) as source_wkw:
187187
num_channels = source_wkw.header.num_channels
188188
source_dtype = source_wkw.header.voxel_type
189-
target_wkw_info.dtype = source_dtype
190189
with open_wkw(
191190
target_wkw_info,
192191
pool_get_lock(),
@@ -410,7 +409,10 @@ def downsample_mag(
410409
interpolation_mode = InterpolationModes[interpolation_mode.upper()]
411410

412411
source_wkw_info = WkwDatasetInfo(path, layer_name, None, source_mag.to_layer_name())
413-
target_wkw_info = WkwDatasetInfo(path, layer_name, None, target_mag.to_layer_name())
412+
with open_wkw(source_wkw_info) as source:
413+
target_wkw_info = WkwDatasetInfo(
414+
path, layer_name, source.header.voxel_type, target_mag.to_layer_name()
415+
)
414416
downsample(
415417
source_wkw_info,
416418
target_wkw_info,

0 commit comments

Comments
 (0)