From 00ef7a84fac017cedf4497082226babff3838404 Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Thu, 23 Mar 2023 15:58:20 +0000
Subject: [PATCH] change(indexing): temporarily revert method
---
src/agora/utils/indexing.py | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/src/agora/utils/indexing.py b/src/agora/utils/indexing.py
index 5ca4494d..8674f0d1 100644
--- a/src/agora/utils/indexing.py
+++ b/src/agora/utils/indexing.py
@@ -114,10 +114,16 @@ def _assoc_indices_to_3d(ndarray: np.ndarray):
"""
result = ndarray
if len(ndarray) and ndarray.ndim > 1:
- result = np.transpose(
- np.hstack((ndarray[:, [0]], ndarray)).reshape(-1, 2, 2),
- axes=[0, 2, 1],
+ columns = np.arange(ndarray.shape[1])
+
+ result = np.stack(
+ (
+ ndarray[:, np.delete(columns, -1)],
+ ndarray[:, np.delete(columns, -2)],
+ ),
+ axis=1,
)
+
return result
--
GitLab