fix: casting original images to numpy.float32

PolusAI · Jan 12, 2024 · 1e6abb5 · 1e6abb5
1 parent bde617b
commit 1e6abb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...y-flatfield-plugin/src/polus/plugins/transforms/images/apply_flatfield/apply_flatfield.py b/...y-flatfield-plugin/src/polus/plugins/transforms/images/apply_flatfield/apply_flatfield.py
@@ -156,7 +156,7 @@ def _unshade_batch(
         images = [f.result() for f in load_futures]
 
     images = [img for _, img in sorted(images, key=operator.itemgetter(0))]
-    img_stack = numpy.stack(images, axis=0)
+    img_stack = numpy.stack(images, axis=0).astype(numpy.float32)
 
     # Apply flatfield correction
     if df_image is not None: