From d13d60af1d7fb9293a82af9e545a1e21133555ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sigbj=C3=B8rn=20Skj=C3=A6ret?= Date: Wed, 18 Mar 2026 23:21:42 +0100 Subject: [PATCH] gguf-py : cleaner way to get the first key (#20727) --- gguf-py/gguf/gguf_writer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 662dda3cf4..57f9fd1a52 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