Skip to content

GitLab

Commit 646dec9e authored by Baber's avatar Baber
Browse files

nit

parent 40885c1b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
lm_eval/api/model.py
View file @ 646dec9e
...@@ -3,7 +3,6 @@ import hashlib ...@@ -3,7 +3,6 @@ import hashlib
import json import json
import logging import logging
import os import os
from typing import TYPE_CHECKING, Optional, Type, TypeVar, Union
from typing import TYPE_CHECKING, Any, Iterable, Optional, Type, TypeVar, Union from typing import TYPE_CHECKING, Any, Iterable, Optional, Type, TypeVar, Union
from tqdm import tqdm from tqdm import tqdm
...@@ -17,10 +16,6 @@ if TYPE_CHECKING: ...@@ -17,10 +16,6 @@ if TYPE_CHECKING:
from lm_eval.api.instance import Instance from lm_eval.api.instance import Instance
if TYPE_CHECKING:
from lm_eval.api.instance import Instance
eval_logger = logging.getLogger(__name__) eval_logger = logging.getLogger(__name__)
T = TypeVar("T", bound="LM") T = TypeVar("T", bound="LM")
...@@ -432,7 +427,9 @@ class TemplateLM(LM): ...@@ -432,7 +427,9 @@ class TemplateLM(LM):
pass pass
@abc.abstractmethod @abc.abstractmethod
def generate_until(self, requests: list[Instance], disable_tqdm: bool = False) -> list[str]: def generate_until(
self, requests: list[Instance], disable_tqdm: bool = False
) -> list[str]:
"""Generate until a stopping sequence. """Generate until a stopping sequence.
Args: Args:
......
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