Update inputs.py (#289)

amorehead · web-flow · commit cac8004b2fad · 2024-09-23T13:39:11.000-07:00
diff --git a/alphafold3_pytorch/inputs.py b/alphafold3_pytorch/inputs.py
@@ -776,6 +776,7 @@ def molecule_to_atom_input(mol_input: MoleculeInput) -> AtomInput:
     i = mol_input
 
     molecules = i.molecules
+    molecule_ids = i.molecule_ids
     atom_lens = i.molecule_token_pool_lens
     extract_atom_feats_fn = i.extract_atom_feats_fn
     extract_atompair_feats_fn = i.extract_atompair_feats_fn
