Update tiktorch/runner/prediction_pipeline/_preprocessing.py

ilastik · Jul 15, 2021 · b32e75c · b32e75c
1 parent 3b1d274
commit b32e75c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tiktorch/runner/prediction_pipeline/_preprocessing.py b/tiktorch/runner/prediction_pipeline/_preprocessing.py
@@ -36,7 +36,7 @@ def zero_mean_unit_variance(tensor: xr.DataArray, axes=None, eps=1.0e-6, mode="p
 
     # monkey patch: maks sure we don't change dtype
     # todo: allow preprocessing to change dtype?
-    return ret.astype(tensor.dtype)
+    return ret.astype("float32")
 
 
 def binarize(tensor: xr.DataArray, *, threshold) -> xr.DataArray: