Fix change of dtype of indices array in ValuationResult

aai-institute · Sep 6, 2024 · 4a53a2e · 4a53a2e
1 parent a6a13e1
commit 4a53a2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pydvl/valuation/result.py b/src/pydvl/valuation/result.py
@@ -263,7 +263,7 @@ def __init__(
 
         if indices is None:
             indices = np.arange(len(self._values), dtype=np.int_)
-        self._indices = np.array(indices, dtype=np.int_, copy=False)
+        self._indices = np.array(indices, dtype=indices.dtype, copy=False)
         self._positions = {idx: pos for pos, idx in enumerate(indices)}
 
         self._sort_positions: NDArray[np.int_] = np.arange(