Merge pull request #83 from mraspaud/fix-palettize-type

Fix palettize dtype for dask arrays
pytroll · Sep 2, 2021 · ea87a27 · ea87a27
2 parents c160f34 + 147e3b5
commit ea87a27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/trollimage/colormap.py b/trollimage/colormap.py
@@ -134,7 +134,7 @@ def palettize(arr, colors, values):
 
 def _palettize_dask(darr, colors, values):
     """Apply a palette to a dask array."""
-    return darr.map_blocks(_palettize, values, dtype=colors.dtype)
+    return darr.map_blocks(_palettize, values, dtype=int)
 
 
 def _palettize(arr, values):

diff --git a/trollimage/tests/test_colormap.py b/trollimage/tests/test_colormap.py
@@ -125,6 +125,7 @@ def test_palettize_dask(self):
         self.assertTrue(np.allclose(channels.compute(), [[0, 1, 2, 3],
                                                          [0, 1, 2, 3],
                                                          [0, 1, 2, 3]]))
+        assert channels.dtype == int
 
     def test_set_range(self):
         """Test set_range."""