pr

stanford-crfm · Oct 14, 2024 · f46690a · f46690a
1 parent f31d994
commit f46690a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/levanter/data/loader.py b/src/levanter/data/loader.py
@@ -205,7 +205,6 @@ def get_local_data_for_leaf(indices: _TensorSliceIndex, leaf_index: int) -> Arra
                 if all(idx == slice(None) for idx in other_indices):
                     return leaf_data
                 else:
-                    # TODO: this doesn't work with named axes
                     return leaf_data[(..., *other_indices)]
 
         def make_global_array_for_leaf(leaf_index, item_leaf_shape: ShapeSpec | NamedShapeSpec):