Commit 6139a5de authored by Baber's avatar Baber
Browse files

add typehint

parent 3285f030
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Generic, Literal, Optional, Tuple, TypeVar from typing import Generic, Literal, Optional, Tuple, TypeVar, Union
from lm_eval.api.types import GenerateInput, LoglikelihoodInput from lm_eval.api.types import GenerateInput, LoglikelihoodInput
...@@ -20,7 +20,7 @@ class Instance(Generic[T]): ...@@ -20,7 +20,7 @@ class Instance(Generic[T]):
metadata: Tuple[Optional[str], Optional[int], Optional[int]] = field( metadata: Tuple[Optional[str], Optional[int], Optional[int]] = field(
default_factory=lambda: (None, None, None) default_factory=lambda: (None, None, None)
) )
resps: list = field(default_factory=list) resps: list[Union[GenerateInput, LoglikelihoodInput]] = field(default_factory=list)
filtered_resps: dict = field(default_factory=dict) filtered_resps: dict = field(default_factory=dict)
# initialized after init # initialized after init
......
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