diff --git a/ndsl/stencils/testing/translate.py b/ndsl/stencils/testing/translate.py index 7e580b7..a25316a 100644 --- a/ndsl/stencils/testing/translate.py +++ b/ndsl/stencils/testing/translate.py @@ -123,6 +123,7 @@ def make_storage_data( axis=axis, names=names_4d, backend=self.stencil_factory.backend, + dtype=array.dtype, ) else: if len(array.shape) == 4: @@ -137,6 +138,7 @@ def make_storage_data( axis=axis, read_only=read_only, backend=self.stencil_factory.backend, + dtype=array.dtype, ) def storage_vars(self):