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