Fix NumericColumn._getrowid() for cases when it is called externally

open-cogsci · Nov 21, 2024 · 5704927 · 5704927
1 parent 11360a0
commit 5704927
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/datamatrix/_datamatrix/_numericcolumn.py b/datamatrix/_datamatrix/_numericcolumn.py
@@ -228,6 +228,8 @@ def _getrowidkey(self, key, dm=None):
 
         if isinstance(key, Index):
             key = key._a
+        if isinstance(key, (tuple, list)):
+            key = np.array(key)
         # argsort and searchsorted are fairly time-consuming operations which
         # need to be performed very often. Therefore we implement a crude
         # but fast caching mechanism.