File size: 960 Bytes
21d2506
 
2e00cda
f2b3434
2e00cda
f2b3434
21d2506
2e00cda
21d2506
f2b3434
2e00cda
f2b3434
 
 
 
 
21d2506
f2b3434
 
d46b2e8
21d2506
d46b2e8
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from dataclasses import dataclass
from typing import Dict, List, Optional, Union

from .artifact import Artifact
from .instructions import InstructionsDict, InstructionsList
from .loaders import Loader
from .normalizers import NormalizeListFields
from .operator import StreamingOperator
from .operators import AddFields, MapInstanceValues
from .task import FormTask
from .templates import TemplatesDict, TemplatesList


class TaskCard(Artifact):
    loader: Loader
    task: FormTask
    preprocess_steps: Optional[List[Union[StreamingOperator, str]]] = None
    templates: Union[TemplatesList, TemplatesDict] = None
    instructions: Union[InstructionsList, InstructionsDict] = None


class ICLCard(Artifact):
    demos_pool_name: str = "demos_pool"
    demos_pool_size: int = None
    demos_field: str = "demos"
    num_demos: int = None
    sampler_type: str = "random"
    instruction_item: Union[str, int] = None
    template_item: Union[str, int] = None