Skip to content

GitLab

Commit ca348256 authored by Baber's avatar Baber
Browse files

nit

parent 684a60ad
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/tasks/ruler/utils.py
View file @ ca348256
...@@ -83,7 +83,7 @@ niah_single_1 = lambda **kwargs: flatten( ...@@ -83,7 +83,7 @@ niah_single_1 = lambda **kwargs: flatten(
for seq in SEQ_LENGTHS for seq in SEQ_LENGTHS
) )
# ruff: noqa # ruff: noqa
niah_single_2 = lambda x: flatten( niah_single_2 = lambda **kwargs: flatten(
generate_samples( generate_samples(
get_haystack(type_haystack="essay"), get_haystack(type_haystack="essay"),
max_seq_length=seq, max_seq_length=seq,
...@@ -91,7 +91,7 @@ niah_single_2 = lambda x: flatten( ...@@ -91,7 +91,7 @@ niah_single_2 = lambda x: flatten(
type_haystack="essay", type_haystack="essay",
type_needle_k="words", type_needle_k="words",
type_needle_v="numbers", type_needle_v="numbers",
TOKENIZER=get_tokenizer(x), TOKENIZER=get_tokenizer(**kwargs),
) )
for seq in SEQ_LENGTHS for seq in SEQ_LENGTHS
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment