Skip to content

Commit 4fdb4e7

Browse files
authored
convert all_clusterings to np
1 parent eed10fd commit 4fdb4e7

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/eschr/tl/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def consensus(n, bg, nprocs):
253253
finish_time = time.perf_counter()
254254
print(f"Consensus clustering finished in {finish_time-start_time} seconds")
255255

256-
return hard_clusters, soft_membership_matrix, all_clusterings_df
256+
return hard_clusters, soft_membership_matrix, all_clusterings_df.to_numpy(dtype=np.uint16)
257257

258258

259259
def consensus_cluster(
@@ -361,7 +361,7 @@ def consensus_cluster(
361361
)
362362

363363
# Obtain consensus from ensemble
364-
hard_clusters, soft_membership_matrix, all_clusterings_df = consensus(
364+
hard_clusters, soft_membership_matrix, all_clusterings = consensus(
365365
n=bipartite.shape[0], bg=bipartite, nprocs=nprocs
366366
)
367367

@@ -375,7 +375,7 @@ def consensus_cluster(
375375
adata.obs["uncertainty_score"] = 1 - np.max(soft_membership_matrix, axis=1)
376376
adata.obsm["bipartite"] = bipartite
377377
if return_multires:
378-
adata.obsm["multiresolution_clusters"] = all_clusterings_df
378+
adata.obsm["multiresolution_clusters"] = all_clusterings
379379

380380
time_per_iter = time.time() - start_time
381381
print("Full runtime: " + str(time_per_iter))

0 commit comments

Comments
 (0)