minor

beyondpie · Jan 24, 2024 · 8b52bfc · 8b52bfc
1 parent f8e9be1
commit 8b52bfc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/snapatac2-python/snapatac2/tools/_embedding.py b/snapatac2-python/snapatac2/tools/_embedding.py
@@ -104,7 +104,9 @@ def umap(
     if use_dims is not None:
         data = data[:, :use_dims] if isinstance(use_dims, int) else data[:, use_dims]
 
-    umap = UMAP(random_state=random_state, n_components=n_comps).fit_transform(data)
+    umap = UMAP(random_state=random_state,
+                n_components=n_comps,
+                **kwargs).fit_transform(data)
     if inplace:
         adata.obsm["X_" + key_added] = umap
     else: