feat: handle slice based access in get_column_values

Renumics · Sep 21, 2023 · 50488f9 · 50488f9
1 parent ec12d45
commit 50488f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/renumics/spotlight_plugins/core/huggingface_datasource.py b/renumics/spotlight_plugins/core/huggingface_datasource.py
@@ -87,8 +87,8 @@ def get_column_values(
             if indices == slice(None):
                 raw_values = self._dataset.data[column_name]
             else:
-                # TODO: handle slices
-                assert False
+                actual_indices = list(range(len(self._dataset)))[indices]
+                raw_values = self._dataset.data[column_name].take[actual_indices]
         else:
             raw_values = self._dataset.data[column_name].take(indices)