fixed docs

raphael-group · Nov 6, 2024 · 56cd3d4 · 56cd3d4
1 parent f75075a
commit 56cd3d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/source/notebooks/paste_tutorial.ipynb b/docs/source/notebooks/paste_tutorial.ipynb
@@ -368,7 +368,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "slices = get_common_genes(slices)\n",
+    "slices, _ = get_common_genes(slices)\n",
     "\n",
     "b = []\n",
     "for i in range(len(slices)):\n",

diff --git a/src/paste3/paste.py b/src/paste3/paste.py
@@ -330,8 +330,7 @@ def center_align(
         device=device,
     )
 
-    if pi_inits is None:
-        pis = [None for _ in slices]
+    pis = [None for _ in slices] if pi_inits is None else pi_inits
 
     # Initialize center_slice
     center_slice = AnnData(np.dot(feature_matrix, coeff_matrix))