From f46690a991c8960e1c49e36d1ada813055ae6eb0 Mon Sep 17 00:00:00 2001
From: David Hall <dlwh@stanford.edu>
Date: Mon, 14 Oct 2024 12:51:43 -0700
Subject: [PATCH] pr

---
 src/levanter/data/loader.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/levanter/data/loader.py b/src/levanter/data/loader.py
index c93043dec..70af3b6f4 100644
--- 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):