@@ -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