From 7e34bc812ae899acb4c51da828a098afd8a938ca Mon Sep 17 00:00:00 2001
From: Fabian Zills <46721498+PythonFZ@users.noreply.github.com>
Date: Wed, 4 Dec 2024 18:17:15 +0100
Subject: [PATCH] fix `np.generic` type issue

---
 apax/nodes/selection.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apax/nodes/selection.py b/apax/nodes/selection.py
index 7e40cc8b..f03fbe42 100644
--- a/apax/nodes/selection.py
+++ b/apax/nodes/selection.py
@@ -174,7 +174,7 @@ def select_atoms(self, atoms_lst: typing.List[ase.Atoms]) -> typing.List[int]:
         self._get_distances_plot(distances, last_selected)
         self._get_pca_plot(features_train, features_selection, features_remaining)
         self._get_selection_plot(atoms_lst, ranking)
-        return list(ranking)
+        return [int(x) for x in ranking]
 
     def _get_selection_plot(
         self, atoms_lst: typing.List[ase.Atoms], indices: typing.List[int]