Fix conversion from sparse matrix to dense array

flatironinstitute · Aug 10, 2024 · 6fe91e6 · 6fe91e6
1 parent f32e8ad
commit 6fe91e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caiman/source_extraction/cnmf/map_reduce.py b/caiman/source_extraction/cnmf/map_reduce.py
@@ -454,7 +454,7 @@ def run_CNMF_patches(file_name, shape, params, gnb=1, dview=None,
         YrA_tot *= nA[:, None]
         nB = np.ravel(np.sqrt(B_tot.power(2).sum(0)))
         B_tot /= nB
-        B_tot = np.array(B_tot, dtype=np.float32)
+        B_tot = B_tot.toarray().astype(np.float32)
 #        B_tot = scipy.sparse.coo_matrix(B_tot)
         F_tot *= nB[:, None]