From 9ee20aa8058c32e6188c354cdbaaa0c35f853050 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Wed, 5 Oct 2022 17:26:20 +0100
Subject: [PATCH] fix(assoc): set correct dimension

---
 src/agora/utils/association.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/agora/utils/association.py b/src/agora/utils/association.py
index 13b9cbe4..656aa787 100644
--- a/src/agora/utils/association.py
+++ b/src/agora/utils/association.py
@@ -71,7 +71,7 @@ def validate_association(
     if association.ndim < 3:
         # Reshape into 3-D array for broadcasting if neded
         association = np.stack(
-            (association[:, [0, 1]], association[:, [0, 2]]), axis=2
+            (association[:, [0, 1]], association[:, [0, 2]]), axis=1
         )
 
     # Compare existing association with available indices
-- 
GitLab