diff --git a/scarf/datastore/mapping_datastore.py b/scarf/datastore/mapping_datastore.py
index e4a42f8..c36978f 100644
--- a/scarf/datastore/mapping_datastore.py
+++ b/scarf/datastore/mapping_datastore.py
@@ -5,7 +5,13 @@
 from dask import array as daskarr
 from loguru import logger
 from scipy.sparse import csr_matrix
-from ..utils import show_dask_progress, clean_array, tqdmbar, controlled_compute, system_call
+from ..utils import (
+    show_dask_progress,
+    clean_array,
+    tqdmbar,
+    controlled_compute,
+    system_call,
+)
 from ..assay import Assay, RNAassay
 from ..writers import create_zarr_dataset
 from .graph_datastore import GraphDataStore
diff --git a/scarf/knn_utils.py b/scarf/knn_utils.py
index 5708bc8..35d7cd5 100644
--- a/scarf/knn_utils.py
+++ b/scarf/knn_utils.py
@@ -82,7 +82,11 @@ def smoothen_dists(store, z_idx, z_dist, lc: float, bw: float, chunk_size: int):
 
     n_cells, n_neighbors = z_idx.shape
     zge = create_zarr_dataset(
-        store, f"edges", (chunk_size * n_neighbors,), ("u8", "u8"), (n_cells * n_neighbors, 2)
+        store,
+        f"edges",
+        (chunk_size * n_neighbors,),
+        ("u8", "u8"),
+        (n_cells * n_neighbors, 2),
     )
     zgw = create_zarr_dataset(
         store, f"weights", (chunk_size * n_neighbors,), "f8", (n_cells * n_neighbors,)