use name output for list iter __next__()

Signed-off-by: Brian Grenier <[email protected]>
modularml · Jan 14, 2025 · 29d992f · 29d992f
1 parent ab8734f
commit 29d992f
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/stdlib/src/collections/list.mojo b/stdlib/src/collections/list.mojo
@@ -60,16 +60,14 @@ struct _ListIter[
     fn __iter__(self) -> Self:
         return self
 
-    fn __next__(
-        mut self,
-    ) -> Pointer[T, list_origin]:
+    fn __next__(mut self, out p: Pointer[T, list_origin]):
         @parameter
         if forward:
+            p = Pointer.address_of(self.src[][self.index])
             self.index += 1
-            return Pointer.address_of(self.src[][self.index - 1])
         else:
             self.index -= 1
-            return Pointer.address_of(self.src[][self.index])
+            p = Pointer.address_of(self.src[][self.index])
 
     @always_inline
     fn __has_next__(self) -> Bool: