Skip to content

Commit 3ed7244

Browse files
committed
style: ruff format on nli.py + _nli_export.py
Co-Authored-By: Arcane Sapience <protoscience@anulum.li>
1 parent 566b2e5 commit 3ed7244

2 files changed

Lines changed: 7 additions & 2 deletions

File tree

src/director_ai/core/scoring/_nli_export.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,9 @@ def export_onnx(
124124

125125
rev = _resolve_revision(model_name, revision)
126126
model = ORTModelForSequenceClassification.from_pretrained(
127-
model_name, export=True, revision=rev,
127+
model_name,
128+
export=True,
129+
revision=rev,
128130
)
129131
model.save_pretrained(output_dir)
130132
AutoTokenizer.from_pretrained(model_name, revision=rev).save_pretrained(output_dir)

src/director_ai/core/scoring/nli.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ def _resolve_revision(model_name: str, revision: str | None = None) -> str | Non
8080
return revision
8181
return MODEL_REGISTRY.get(model_name)
8282

83+
8384
# FactCG instruction template (NAACL 2025, derenlei/FactCG)
8485
_FACTCG_TEMPLATE = (
8586
"{text_a}\n\nChoose your answer: based on the paragraph above "
@@ -533,7 +534,9 @@ def _ensure_minicheck(self) -> bool:
533534
)
534535
config.problem_type = "single_label_classification"
535536
inf.tokenizer = AutoTokenizer.from_pretrained(
536-
ckpt, use_fast=True, cache_dir=self._cache_dir,
537+
ckpt,
538+
use_fast=True,
539+
cache_dir=self._cache_dir,
537540
revision=mc_rev,
538541
)
539542
inf.model = AutoModelForSequenceClassification.from_pretrained(

0 commit comments

Comments
 (0)