File size: 810 Bytes
2e00cda
 
f2b3434
2e00cda
f2b3434
2e00cda
f2b3434
2e00cda
f2b3434
 
 
 
 
 
 
 
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
from typing import List, Optional, Union

from .artifact import Artifact
from .instructions import InstructionsDict, InstructionsList
from .loaders import Loader
from .operator import StreamingOperator
from .task import FormTask
from .templates import TemplatesDict, TemplatesList


class TaskCard(Artifact):
    loader: Loader
    task: FormTask
    preprocess_steps: Optional[List[StreamingOperator]] = 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