From 4f14e97effff70aa102bb330d59b75f3244af3ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Wed, 15 Mar 2023 20:43:59 +0000
Subject: [PATCH] change(picker): import agora tools

---
 src/postprocessor/core/reshapers/picker.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/postprocessor/core/reshapers/picker.py b/src/postprocessor/core/reshapers/picker.py
index 4132cf8d..1cc0bdc8 100644
--- a/src/postprocessor/core/reshapers/picker.py
+++ b/src/postprocessor/core/reshapers/picker.py
@@ -5,9 +5,9 @@ import pandas as pd
 
 from agora.abc import ParametersABC
 from agora.io.cells import Cells
-
 from agora.utils.association import validate_association
-from agora.utils.kymograph import drop_mother_label, get_index_as_np
+from agora.utils.cast import _str_to_int
+from agora.utils.kymograph import drop_mother_label
 from postprocessor.core.lineageprocess import LineageProcess
 
 
@@ -100,7 +100,7 @@ class Picker(LineageProcess):
             self._log(f"No lineage assignment")
             indices = np.array([])
 
-        return np.array(list(indices))
+        return np.array(tuple(map(_str_to_int, x)) for x in indices)
 
     def switch_case(
         self,
-- 
GitLab