Unverified Commit 5ce8e045 authored by Fengzhe Zhou's avatar Fengzhe Zhou Committed by GitHub
Browse files

[Fix] Fix type hint in IFEval (#915)

parent 53fe788d
......@@ -16,7 +16,7 @@
# limitations under the License.
import dataclasses
from typing import Dict, Optional, Union
from typing import Dict, List, Optional, Union
from absl import flags
......@@ -43,18 +43,18 @@ _OUTPUT_DIR = flags.DEFINE_string(
@dataclasses.dataclass
class InputExample:
key: int
instruction_id_list: list[str]
instruction_id_list: List[str]
prompt: str
kwargs: list[Dict[str, Optional[Union[str, int]]]]
kwargs: List[Dict[str, Optional[Union[str, int]]]]
@dataclasses.dataclass
class OutputExample:
instruction_id_list: list[str]
instruction_id_list: List[str]
prompt: str
response: str
follow_all_instructions: bool
follow_instruction_list: list[bool]
follow_instruction_list: List[bool]
def test_instruction_following_strict(
......
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