Add missing bracket

f1402490 · Gustaf Ahdritz · 4e934f74 · f1402490
Commit f1402490 authored Jun 21, 2022 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/model/structure_module.py openfold/model/structure_module.py +1 -1

No files found.
--- a/openfold/model/structure_module.py
+++ b/openfold/model/structure_module.py
@@ -233,7 +233,7 @@ class InvariantPointAttention(nn.Module):
        r: Rigid,
        mask: torch.Tensor,
        _offload_inference: bool = False,
-        _z_reference_list: Optional[Sequence[torch.Tensor] = None,
+        _z_reference_list: Optional[Sequence[torch.Tensor]] = None,
    ) -> torch.Tensor:
        """
        Args: