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