Nit

huggingface · Apr 16, 2024 · 5082217 · 5082217
1 parent 0ec973d
commit 5082217
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/src/datasets/formatting/formatting.py b/src/datasets/formatting/formatting.py
@@ -75,11 +75,6 @@ def _query_table_with_indices_mapping(
     if isinstance(key, Iterable):
         return _query_table(table, [indices.fast_slice(i, 1).column(0)[0].as_py() for i in key])
 
-    try:
-        return operator.index(key)
-    except TypeError:
-        pass
-
     _raise_bad_key_type(key)
 
 
@@ -105,11 +100,6 @@ def _query_table(table: Table, key: Union[int, slice, range, str, Iterable]) ->
         # don't use pyarrow.Table.take even for pyarrow >=1.0 (see https://issues.apache.org/jira/browse/ARROW-9773)
         return table.fast_gather(key % table.num_rows)
 
-    try:
-        return operator.index(key)
-    except TypeError:
-        pass
-
     _raise_bad_key_type(key)