Skip to content

Commit a84d263

Browse files
Augustin-Zidekcopybara-github
authored andcommitted
Add early validation that RNG seeds are uint32s
Inspired by #685 PiperOrigin-RevId: 936470606 Change-Id: I415b47de84c6337e4dff87ea2996dc7020974031
1 parent 7b197fe commit a84d263

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/alphafold3/common/folding_input.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -980,6 +980,11 @@ def __post_init__(self):
980980
if not self.rng_seeds:
981981
raise ValueError('Input must have at least one RNG seed.')
982982

983+
if any(seed < 0 or seed > 2**32 - 1 for seed in self.rng_seeds):
984+
raise ValueError(
985+
f'RNG seeds must be 32-bit unsigned integers, got {self.rng_seeds}'
986+
)
987+
983988
if not self.name.strip() or not self.sanitised_name():
984989
raise ValueError(
985990
'Input name must be non-empty and contain at least one valid'

0 commit comments

Comments
 (0)