From 83c4be2ec188b79e664256fde837f7ca7b318794 Mon Sep 17 00:00:00 2001
From: Augustin Zidek <augustinzidek@google.com>
Date: Mon, 18 Nov 2024 16:23:20 +0000
Subject: [PATCH] Don't print whole param payload if it is corrupt

PiperOrigin-RevId: 697635175
---
 src/alphafold3/model/params.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/alphafold3/model/params.py b/src/alphafold3/model/params.py
index 59b75f5..2fa3407 100644
--- a/src/alphafold3/model/params.py
+++ b/src/alphafold3/model/params.py
@@ -57,7 +57,7 @@ def _read_record(stream: IO[bytes]) -> tuple[str, str, np.ndarray] | None:
   if not header:
     return None
   if len(header) < header_size:
-    raise RecordError(f'Incomplete header: {header}')
+    raise RecordError(f'Incomplete header: {len(header)=} < {header_size=}')
   (scope_len, name_len, dtype_len, shape_len, arr_buffer_len) = struct.unpack(
       '<5i', header
   )
@@ -65,7 +65,7 @@ def _read_record(stream: IO[bytes]) -> tuple[str, str, np.ndarray] | None:
   payload_size = struct.calcsize(fmt) + arr_buffer_len
   payload = stream.read(payload_size)
   if len(payload) < payload_size:
-    raise RecordError(f'Incomplete payload: {payload}')
+    raise RecordError(f'Incomplete payload: {len(payload)=} < {payload_size=}')
   scope, name, dtype, *shape = struct.unpack_from(fmt, payload)
   scope = scope.decode('utf-8')
   name = name.decode('utf-8')
-- 
GitLab