We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent eb43339 commit 665b750Copy full SHA for 665b750
autointent/modules/scoring/_cnn/textcnn.py
@@ -59,7 +59,7 @@ def load(self, model_path: str) -> None:
59
state_dict = torch.load(model_path)
60
self.load_state_dict(state_dict)
61
62
- def get_config(self) -> -> dict[str, int | list[int] | torch.Tensor | None]:
+ def get_config(self) -> dict[str, int | list[int] | torch.Tensor | None]:
63
return {
64
"vocab_size": self.vocab_size.item(),
65
"n_classes": self.n_classes.item(),
0 commit comments