From 2390b8db0e6e7a5d3d3b954b98e7c6ab39853afd Mon Sep 17 00:00:00 2001
From: Shing Zhan <shing.zhan@gmail.com>
Date: Tue, 12 Sep 2023 16:49:52 +0100
Subject: [PATCH] More explicit

---
 python/tests/beagle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tests/beagle.py b/python/tests/beagle.py
index d197150c6c..170984e982 100644
--- a/python/tests/beagle.py
+++ b/python/tests/beagle.py
@@ -528,7 +528,7 @@ def get_map_alleles(allele_probs):
     """
     assert not np.any(allele_probs < 0), "Allele probabilities have negative values."
     assert not np.any(np.isnan(allele_probs)), "Allele probabilities have NaN values."
-    x = len(allele_probs)
+    x = allele_probs.shape[0]
     imputed_alleles = np.zeros(x, dtype=int)
     # TODO: Vectorise over the imputed markers
     for i in np.arange(x):