Skip to content

Commit 9f0be86

Browse files
committed
fix: fix bug in repr in exact self-attention modul
1 parent 819a29e commit 9f0be86

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cellseg_models_pytorch/modules/self_attention/exact_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,6 @@ def forward(
9494

9595
def __repr__(self) -> str:
9696
"""Add extra info to print."""
97-
s = "ExactSelfAttention(self_attention='{self_attention}', head_dim={head_dim}, num_heads={num_heads})" # noqa: E501
97+
s = "ExactSelfAttention(how='{how}', head_dim={head_dim}, num_heads={num_heads})" # noqa: E501
9898
s = s.format(**self.__dict__)
9999
return s

0 commit comments

Comments
 (0)