Revert "Fix tfidf using the wrong layer"

scverse · Oct 17, 2024 · f9da198 · f9da198
1 parent d52b60d
commit f9da198
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/muon/_atac/preproc.py b/muon/_atac/preproc.py
@@ -96,14 +96,14 @@ def tfidf(
         tf = np.dot(n_peaks, counts)
     else:
         n_peaks = np.asarray(counts.sum(axis=1)).reshape(-1, 1)
-        tf = counts / n_peaks
+        tf = adata.X / n_peaks
 
     if scale_factor is not None and scale_factor != 0 and scale_factor != 1:
         tf = tf * scale_factor
     if log_tf:
         tf = np.log1p(tf)
 
-    idf = np.asarray(adata.shape[0] / counts.sum(axis=0)).reshape(-1)
+    idf = np.asarray(adata.shape[0] / adata.X.sum(axis=0)).reshape(-1)
     if log_idf:
         idf = np.log1p(idf)