nit

bd255f61 · Baber · 06a04113 · bd255f61
Commit bd255f61 authored Feb 05, 2025 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/models/api_models.py lm_eval/models/api_models.py +1 -1

No files found.
--- a/lm_eval/models/api_models.py
+++ b/lm_eval/models/api_models.py
@@ -725,7 +725,7 @@ class TemplateAPI(TemplateLM):

    def simple_async_generate(
        self,
-        requests: Union[List[List[str], list[list[dict]]], List[List[dict]]],
+        requests: Union[List[List[str]], List[List[dict]]],
        gen_kwargs: dict,
    ):
        results = itertools.chain.from_iterable(