Supply the parition as an arg rather than a kwarg

UM-RMRS · Jan 16, 2025 · 00d76c3 · 00d76c3
1 parent 2f04330
commit 00d76c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/raster_tools/rasterize.py b/raster_tools/rasterize.py
@@ -371,11 +371,11 @@ def _rasterize_spatial_matches(
             if not mask:
                 chunk = da.map_blocks(
                     _rasterize_onto_chunk,
+                    part,
                     dtype=target_dtype,
                     chunks=little_like.shape[1:],
                     meta=np.array((), dtype=target_dtype),
                     # func args
-                    gdf=part,
                     transform=little_like.affine,
                     out_dtype=target_dtype,
                     fill=fill,
@@ -385,11 +385,11 @@ def _rasterize_spatial_matches(
             else:
                 chunk = da.map_blocks(
                     _mask_onto_chunk,
+                    part,
                     dtype=U8,
                     chunks=little_like.shape[1:],
                     meta=np.array((), dtype=U8),
                     # Func args
-                    gdf=part,
                     transform=little_like.affine,
                     all_touched=all_touched,
                     invert=mask_invert,