Fix outputs

flatironinstitute · Apr 12, 2024 · 27bfecb · 27bfecb
1 parent c45a0ee
commit 27bfecb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/inferelator_velocity/utils/graph.py b/inferelator_velocity/utils/graph.py
@@ -13,7 +13,13 @@
 try:
     from scanpy.neighbors import _compute_connectivities_umap as _connect_umap
 except ImportError:
-    from scanpy.neighbors._connectivity import umap as _connect_umap
+    from scanpy.neighbors._connectivity import umap as _scanpy_connect
+
+    def _connect_umap(i, d, **kwargs):
+
+        return d, _scanpy_connect(
+            i, d, **kwargs
+        )
 
 
 def compute_neighbors(