Make Dilate arguments JSON serializable (#81)

nens · Nov 19, 2020 · 3ed3b42 · 3ed3b42
1 parent d12eecd
commit 3ed3b42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,7 +4,7 @@ Changelog of dask-geomodeling
 2.3.2 (unreleased)
 ------------------
 
-- Nothing changed yet.
+- Make Dilate arguments JSON serializable. (#81)
 
 
 2.3.1 (2020-11-10)
@@ -621,4 +621,3 @@ Changelog of dask-geomodeling
 - Fixed issue with ciso8601 2.*
 
 - Bumped raster-store dependency to 4.0.0.
-
diff --git a/dask_geomodeling/raster/spatial.py b/dask_geomodeling/raster/spatial.py
@@ -132,7 +132,7 @@ class Dilate(BaseSingle):
 
     def __init__(self, store, values):
         values = np.asarray(values, dtype=store.dtype)
-        super(Dilate, self).__init__(store, values)
+        super().__init__(store, values.tolist())
 
     @property
     def values(self):
@@ -151,7 +151,7 @@ def process(data, values=None):
             return data
         original = data["values"]
         dilated = original.copy()
-        for value in values:
+        for value in np.asarray(values, dtype=original.dtype):
             dilated[ndimage.binary_dilation(original == value)] = value
         dilated = dilated[:, 1:-1, 1:-1]
         return {"values": dilated, "no_data_value": data["no_data_value"]}