Skip to content

Commit dc962a2

Browse files
Nathan Habibclaude
authored andcommitted
Fix pre-existing ruff format drift in README and docs
Unrelated to the credential-leak fixes, but CI's Quality check was already red on main for this before this branch existed - fixing it here so this PR's checks are green. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
1 parent b2b71c6 commit dc962a2

2 files changed

Lines changed: 3 additions & 13 deletions

File tree

README.md

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -145,14 +145,9 @@ MODEL_NAME = "meta-llama/Meta-Llama-3-8B-Instruct"
145145
BENCHMARKS = "gsm8k"
146146

147147
evaluation_tracker = EvaluationTracker(output_dir="./results")
148-
pipeline_params = PipelineParameters(
149-
launcher_type=ParallelismManager.NONE,
150-
max_samples=2
151-
)
148+
pipeline_params = PipelineParameters(launcher_type=ParallelismManager.NONE, max_samples=2)
152149

153-
model = AutoModelForCausalLM.from_pretrained(
154-
MODEL_NAME, device_map="auto"
155-
)
150+
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto")
156151
config = TransformersModelConfig(model_name=MODEL_NAME, batch_size=1)
157152
model = TransformersModel.from_model(model, config)
158153

docs/source/offline-evaluation.md

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,7 @@ from lighteval.tasks.requests import Doc
1818

1919

2020
def local_prompt(line: dict, task_name: str) -> Doc:
21-
return Doc(
22-
task_name=task_name,
23-
query=line["question"],
24-
choices=line["choices"],
25-
gold_index=line["answer"]
26-
)
21+
return Doc(task_name=task_name, query=line["question"], choices=line["choices"], gold_index=line["answer"])
2722

2823

2924
local_data = Path(__file__).parent / "samples" / "faq.jsonl"

0 commit comments

Comments
 (0)