Skip to content

Commit 5f9ac06

Browse files
update to_cupy
Signed-off-by: Yiheng Wang <[email protected]>
1 parent a62b1dc commit 5f9ac06

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

monai/data/image_reader.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -986,9 +986,8 @@ def get_data(self, img) -> tuple[np.ndarray, dict]:
986986
else:
987987
header[MetaKeys.ORIGINAL_CHANNEL_DIM] = self.channel_dim
988988
_copy_compatible_dict(header, compatible_meta)
989-
if self.to_gpu:
990-
return _stack_images(img_array, compatible_meta, to_cupy=True), compatible_meta
991-
return _stack_images(img_array, compatible_meta), compatible_meta
989+
990+
return _stack_images(img_array, compatible_meta, to_cupy=self.to_gpu), compatible_meta
992991

993992
def _get_meta_dict(self, img) -> dict:
994993
"""

0 commit comments

Comments
 (0)