Skip to content

Commit 85d500b

Browse files
committed
Fix mypy typing
1 parent 7d1d5d0 commit 85d500b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

autointent/modules/scoring/_ptuning/ptuning.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import torch
99
from datasets import Dataset
1010
from peft import PromptEncoderConfig, get_peft_model
11-
from transformers import (
11+
from transformers import ( # type: ignore[attr-defined]
1212
AutoModelForSequenceClassification,
1313
AutoTokenizer,
1414
DataCollatorWithPadding,
@@ -188,15 +188,15 @@ def tokenize_function(examples: dict[str, Any]) -> dict[str, Any]:
188188

189189
data_collator = DataCollatorWithPadding(tokenizer=self._tokenizer)
190190

191-
trainer = Trainer(
191+
trainer = Trainer( # type: ignore[no-untyped-call]
192192
model=self._model,
193193
args=training_args,
194194
train_dataset=tokenized_dataset,
195195
processing_class=self._tokenizer,
196196
data_collator=data_collator,
197197
)
198198

199-
trainer.train()
199+
trainer.train() # type: ignore[attr-defined]
200200

201201
self._model.eval()
202202

0 commit comments

Comments
 (0)