Refactor PPCA a bit

cwindolf · Jan 10, 2025 · baa060c · baa060c
1 parent d1237b7
commit baa060c
Show file tree

Hide file tree

Showing 3 changed files with 95 additions and 181 deletions.
diff --git a/src/dartsort/cluster/gaussian_mixture.py b/src/dartsort/cluster/gaussian_mixture.py
@@ -3108,7 +3108,7 @@ def noise_whiten(
         nu = z.new_zeros((nbids.numel(), *z.shape[1:]))
     for j, nbid in enumerate(nbids):
         nbchans = neighborhoods.neighborhoods[nbid]
-        nbvalid = nbchans < noise.n_channels
+        nbvalid = neighborhoods.valid_mask(nbid)
         nbchans = nbchans[nbvalid]
         innb = sp.neighborhood_ids == nbid
         nbcov = noise.marginal_covariance(