Skip to content

Commit 77f9751

Browse files
committed
Update _dump_tools.py
1 parent 22954c9 commit 77f9751

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autointent/_dump_tools.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def dump(obj: Any, path: Path) -> None: # noqa: ANN401, C901, PLR0912, PLR0915
118118
model_path = path / Dumper.hf_models / key
119119
model_path.mkdir(parents=True, exist_ok=True)
120120
try:
121-
val.save_pretrained(model_path) # type: ignore[operator]
121+
val.save_pretrained(model_path)
122122
class_info = {"module": val.__class__.__module__, "name": val.__class__.__name__}
123123
with (model_path / "class_info.json").open("w") as f:
124124
json.dump(class_info, f)
@@ -129,7 +129,7 @@ def dump(obj: Any, path: Path) -> None: # noqa: ANN401, C901, PLR0912, PLR0915
129129
tokenizer_path = path / Dumper.hf_tokenizers / key
130130
tokenizer_path.mkdir(parents=True, exist_ok=True)
131131
try:
132-
val.save_pretrained(tokenizer_path) # type: ignore[operator]
132+
val.save_pretrained(tokenizer_path)
133133
class_info = {"module": val.__class__.__module__, "name": val.__class__.__name__}
134134
with (tokenizer_path / "class_info.json").open("w") as f:
135135
json.dump(class_info, f)

0 commit comments

Comments
 (0)