Skip to content

GitLab

Commit 37f17c3a authored by Baber's avatar Baber
Browse files

nit

parent ee4252ff
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/tasks/ruler/prepare.py
View file @ 37f17c3a
...@@ -3,7 +3,7 @@ import random ...@@ -3,7 +3,7 @@ import random
import uuid import uuid
from linecache import cache from linecache import cache
from functools import lru_cache from functools import lru_cache
from typing import List from typing import List, Union
import numpy as np import numpy as np
import wonderwords import wonderwords
...@@ -97,7 +97,7 @@ def generate_random(type_needle: str): ...@@ -97,7 +97,7 @@ def generate_random(type_needle: str):
def generate_input_output( def generate_input_output(
num_haystack: int, num_haystack: int,
haystack: list[str] | str, haystack: Union[list[str], str],
*, *,
type_haystack: str, type_haystack: str,
num_needle_k: int, num_needle_k: int,
......
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