From: Sigbjørn Skjæret <redacted>
Date: Wed, 18 Mar 2026 22:21:42 +0000 (+0100)
Subject: gguf-py : cleaner way to get the first key (#20727)
X-Git-Tag: upstream/0.0.8611~196
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=d13d60af1d7fb9293a82af9e545a1e21133555ce;p=pkg%2Fggml%2Fsources%2Fllama.cpp

gguf-py : cleaner way to get the first key (#20727)
---

diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py
index 662dda3cf..57f9fd1a5 100644
--- a/gguf-py/gguf/gguf_writer.py
+++ b/gguf-py/gguf/gguf_writer.py
@@ -425,8 +425,7 @@ class GGUFWriter:
         fout = self.fout[file_id]
 
         # pop the first tensor info
-        # TODO: cleaner way to get the first key
-        first_tensor_name = [name for name, _ in zip(self.tensors[file_id].keys(), range(1))][0]
+        first_tensor_name = next(iter(self.tensors[file_id]))
         ti = self.tensors[file_id].pop(first_tensor_name)
         assert ti.nbytes == tensor.nbytes