pop outside iterator

Signed-off-by: jayzhan211 <jayzhan211@gmail.com>
modularml · May 22, 2024 · 4fa4599 · 4fa4599
1 parent 4700fb7
commit 4fa4599
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/stdlib/src/collections/dict.mojo b/stdlib/src/collections/dict.mojo
@@ -750,11 +750,17 @@ struct Dict[K: KeyElement, V: CollectionElement](
             "KeyError" if the dictionary is empty.
         """
 
+        var key = Optional[K](None)
+        var val = Optional[V](None)
+
         for item in reversed(self.items()):
-            var key = item[].key
-            var value = item[].value
-            _ = self.pop(key)
-            return DictEntry[K, V](key, value)
+            key = Optional(item[].key)
+            val = Optional(item[].value)
+            break
+
+        if key:
+            _ = self.pop(key.value()[])
+            return DictEntry[K, V](key.value()[], val.value()[])
 
         raise "KeyError"