Add CPU unit tests and tighten type-checking (foundry rigid.py + rf3 helpers)#317

Closed

lyskov-ai wants to merge 8 commits into

RosettaCommons:productionfrom

lyskov-ai:0034-foundry-rigid-mypy-strict-and-tests

Commits on Jun 9, 2026

test(rf3): add CPU unit tests for closed-form loss gradients and util_module

lyskov
and
lyskov-ai
committed
test(rf3): add CPU unit tests for layer_utils shape helpers

lyskov
and
lyskov-ai
committed
test(rf3): add CPU unit tests for the diffusion sampler and outer-product layer

lyskov
and
lyskov-ai
committed
test(rf3): add CPU unit tests for the mlff and triangle-attention layers

lyskov
and
lyskov-ai
committed
test(rf3): add CPU unit tests for the af3_losses symmetry-resolution helpers

lyskov
and
lyskov-ai
committed
test(rf3): add CPU unit tests for chiral and iPTM metric orchestration

lyskov
and
lyskov-ai
committed
chore(rf3): move paired_msa mypy suppression into the file

lyskov
and
lyskov-ai
committed

Commits on Jun 10, 2026

chore(foundry): annotate rigid.py for strict mypy and add geometry unit tests

lyskov
and
lyskov-ai
committed