Fix mypy errors: add type: ignore for dask.config.set() calls

Author: khushthecoder

malariagen_data/anoph/snp_data.py

@@ -610,13 +610,13 @@ def snp_genotypes(
             )
 
             # Filter the Dask array using the boolean array.
-            with dask.config.set(**{"array.slicing.split_native_chunks": False}):
+            with dask.config.set({"array.slicing.split_native_chunks": False}):  # type: ignore[arg-type]
                 d = da.compress(loc_samples, d, axis=1)
 
         # Apply the sample_indices, if there are any.
         # Note: this might need to apply to the result of an internal sample_query, e.g. `is_surveillance == True`.
         if sample_indices is not None:
-            with dask.config.set(**{"array.slicing.split_native_chunks": False}):
+            with dask.config.set({"array.slicing.split_native_chunks": False}):  # type: ignore[arg-type]
                 d = da.take(d, sample_indices, axis=1)
 
         return d
@@ -1020,7 +1020,7 @@ def _locate_site_class(
                 chunks=chunks,
             )
             idx = (pos - 1).compute()
-            with dask.config.set(**{"array.slicing.split_native_chunks": False}):
+            with dask.config.set({"array.slicing.split_native_chunks": False}):  # type: ignore[arg-type]
                 loc_ann = da.take(loc_ann, idx, axis=0)
 
             # Compute site selection.

malariagen_data/util.py

@@ -411,7 +411,7 @@ def _da_compress(
     # Apply the indexing operation, suppressing the dask performance warning
     # about split_native_chunks. This config is scoped here rather than at
     # module level to avoid silently modifying global dask configuration.
-    with dask.config.set(**{"array.slicing.split_native_chunks": False}):
+    with dask.config.set({"array.slicing.split_native_chunks": False}):  # type: ignore[arg-type]
         v = da.compress(indexer, data, axis=axis)
 
     # Need to compute chunks sizes in order to know dimension sizes;