Skip to content

Commit 065e4aa

Browse files
committed
fix ruff
1 parent d6a45e1 commit 065e4aa

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

autointent/_dump_tools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def make_subdirectories(path: Path, exists_ok: bool = False) -> None:
6767
path / Dumper.hf_models,
6868
path / Dumper.hf_tokenizers,
6969
path / Dumper.torch_models,
70-
path / Dumper.containers
70+
path / Dumper.containers,
7171
path / Dumper.ptuning_models,
7272
]
7373
for subdir in subdirectories:

autointent/modules/scoring/_cnn/textcnn.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,12 @@ def load(self, model_path: str) -> None:
7575

7676
def get_config(self) -> dict:
7777
return {
78-
'vocab_size': self.vocab_size.item(),
79-
'n_classes': self.n_classes.item(),
80-
'embed_dim': self.embed_dim.item(),
81-
'kernel_sizes': self.kernel_sizes.tolist(),
82-
'num_filters': self.num_filters.item(),
83-
'dropout': self.dropout_rate.item(),
84-
'padding_idx': self.padding_idx.item(),
85-
'pretrained_embs': self.pretrained_embs,
86-
}
78+
"vocab_size": self.vocab_size.item(),
79+
"n_classes": self.n_classes.item(),
80+
"embed_dim": self.embed_dim.item(),
81+
"kernel_sizes": self.kernel_sizes.tolist(),
82+
"num_filters": self.num_filters.item(),
83+
"dropout": self.dropout_rate.item(),
84+
"padding_idx": self.padding_idx.item(),
85+
"pretrained_embs": self.pretrained_embs,
86+
}

0 commit comments

Comments
 (0)