Fix bug

tskit-dev · Jan 13, 2024 · 90e0b10 · 90e0b10
1 parent 68607be
commit 90e0b10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tests/beagle.py b/python/tests/beagle.py
@@ -76,7 +76,7 @@ def convert_to_genetic_map_position(pos, genetic_map=None):
     xMa = np.zeros(len(pos), dtype=np.float32)
     bMa = np.zeros(len(pos), dtype=np.float32)
     fbMfa = np.zeros(len(pos), dtype=np.float32)
-    for i in range(pos):
+    for i in range(len(pos)):
         xMa[i] = pos[i] - base_pos[left_idx[i]]
         bMa[i] = base_pos[left_idx[i] + 1] - base_pos[left_idx[i]]
         fbMfa[i] = gen_pos[left_idx[i] + 1] - gen_pos[left_idx[i]]

diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -42,7 +42,7 @@ def convert_to_genetic_map_position(pos, genetic_map_pos=None, genetic_map_cm=No
     xMa = np.zeros(len(pos), dtype=np.float32)
     bMa = np.zeros(len(pos), dtype=np.float32)
     fbMfa = np.zeros(len(pos), dtype=np.float32)
-    for i in range(pos):
+    for i in range(len(pos)):
         xMa[i] = pos[i] - base_pos[left_idx[i]]
         bMa[i] = base_pos[left_idx[i] + 1] - base_pos[left_idx[i]]
         fbMfa[i] = gen_pos[left_idx[i] + 1] - gen_pos[left_idx[i]]