File size: 940 Bytes
21d2506 2e00cda f2b3434 fb4afb9 2e00cda f2b3434 21d2506 2e00cda 21d2506 f2b3434 2e00cda f2b3434 21d2506 fb4afb9 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 31 |
from dataclasses import dataclass
from typing import Dict, List, Optional, Union
from .artifact import Artifact
from .collections import Collection
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: Collection = None
instructions: Collection = 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
|