From adad0efb080c9f16ba203b9561db4ecbe9f2381e Mon Sep 17 00:00:00 2001 From: Kemal Yaylali Date: Tue, 29 Sep 2026 19:54:24 +0100 Subject: [PATCH] train: loop variable typing (mypy) --- src/simurg/train/clusters.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/simurg/train/clusters.py b/src/simurg/train/clusters.py index 6a87e62..5b06a09 100644 --- a/src/simurg/train/clusters.py +++ b/src/simurg/train/clusters.py @@ -106,10 +106,10 @@ def fit_block(theta: F32, r: F32, use: npt.NDArray[np.bool_], haploid: npt.NDArr # a locus with a single cluster is homozygous: an AB-labelled lone cluster moves to AA or BB by position lone_ab = keep[:, 1] & ~keep[:, 0] & ~keep[:, 2] if lone_ab.any(): - for i in np.flatnonzero(lone_ab).tolist(): - hom = 0 if m[i, 1] < 0.5 else 2 - assign[i, assign[i] == 1] = hom - keep[i, 1], keep[i, hom] = False, True + for j in np.flatnonzero(lone_ab).tolist(): + hom = 0 if m[j, 1] < 0.5 else 2 + assign[j, assign[j] == 1] = hom + keep[j, 1], keep[j, hom] = False, True counts = np.stack([(assign == k).sum(axis=1) for k in range(3)], axis=1) tm = np.full((L, 3), np.nan) td = np.full((L, 3), np.nan)