Skip to content

Commit 665b750

Browse files
committed
Update textcnn.py
1 parent eb43339 commit 665b750

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autointent/modules/scoring/_cnn/textcnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def load(self, model_path: str) -> None:
5959
state_dict = torch.load(model_path)
6060
self.load_state_dict(state_dict)
6161

62-
def get_config(self) -> -> dict[str, int | list[int] | torch.Tensor | None]:
62+
def get_config(self) -> dict[str, int | list[int] | torch.Tensor | None]:
6363
return {
6464
"vocab_size": self.vocab_size.item(),
6565
"n_classes": self.n_classes.item(),

0 commit comments

Comments
 (0)