Update inputs.py (#299)

lucidrains · Oct 1, 2024 · 7258c56 · 7258c56
1 parent 0b70ab9
commit 7258c56
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/alphafold3_pytorch/inputs.py b/alphafold3_pytorch/inputs.py
@@ -3121,8 +3121,7 @@ def pdb_input_to_molecule_input(
 
     if (
         i.distillation
-        and not_exists(i.chains)
-        or (exists(i.chains) and not (exists(i.chains[0]) or exists(i.chains[1])))
+        and (not_exists(i.chains) or (exists(i.chains) and not (exists(i.chains[0]) or exists(i.chains[1]))))
     ):
         chain_id_1, chain_id_2 = i.chains if exists(i.chains) else (None, None)