Merge branch 'add-trimming-to-bbknn' into faster-trim

scverse · Nov 4, 2024 · b3b8b5d · b3b8b5d
2 parents 01b374b + 8486c2d
commit b3b8b5d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_neighbors.py b/tests/test_neighbors.py
@@ -95,8 +95,8 @@ def test_trimming():
     cnts_gpu = X_to_GPU(adata.obsp["connectivities"]).astype(np.float32)
     cnts_cpu = adata.obsp["connectivities"].astype(np.float32)
 
-    cnts_cpu = trimming_cpu(cnts_gpu, 5)
-    cnts_gpu = trimming_gpu(cnts_cpu, 5)
+    cnts_cpu = trimming_cpu(cnts_cpu, 5)
+    cnts_gpu = trimming_gpu(cnts_gpu, 5)
 
     cp.testing.assert_array_equal(cnts_cpu.data, cnts_gpu.data)
     cp.testing.assert_array_equal(cnts_cpu.indices, cnts_gpu.indices)