Skip to content

Commit 910e075

Browse files
authored
Update alphafold3.py (#283)
1 parent 492b4b8 commit 910e075

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

alphafold3_pytorch/alphafold3.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2782,7 +2782,7 @@ def forward(
27822782
return_loss_breakdown = False,
27832783
single_structure_input=False,
27842784
verbose=None,
2785-
filepaths: List[str] | None = None,
2785+
filepaths: List[str] | Tuple[str] | None = None,
27862786
) -> ElucidatedAtomDiffusionReturn:
27872787
verbose = default(verbose, self.verbose)
27882788

@@ -6485,7 +6485,7 @@ def forward(
64856485
distance_labels: Int['b n n'] | Int['b m m'] | None = None,
64866486
resolved_labels: Int['b m'] | None = None,
64876487
resolution: Float[' b'] | None = None,
6488-
token_constraints: Int['b n n dac'] | None = None,
6488+
token_constraints: Float['b n n dac'] | None = None,
64896489
return_loss_breakdown = False,
64906490
return_loss: bool = None,
64916491
return_all_diffused_atom_pos: bool = False,
@@ -6499,7 +6499,7 @@ def forward(
64996499
max_conf_resolution: float = 4.0,
65006500
hard_validate: bool = False,
65016501
verbose: bool | None = None,
6502-
filepaths: List[str] | None = None
6502+
filepaths: List[str] | Tuple[str] | None = None
65036503
) -> (
65046504
Float['b m 3'] |
65056505
List[Structure] |

0 commit comments

Comments
 (0)