metric / card.py
Elron's picture
Upload card.py with huggingface_hub
370e1f5
raw
history blame
819 Bytes
from typing import List, Optional, Union
from .artifact import Artifact
from .collections import Collection
from .dataclass import OptionalField
from .loaders import Loader
from .operator import StreamingOperator
from .splitters import RandomSampler, Sampler
from .task import FormTask
class TaskCard(Artifact):
loader: Loader
task: FormTask
preprocess_steps: List[StreamingOperator] = None
templates: Collection = None
instructions: Collection = None
sampler: Sampler = OptionalField(default_factory=RandomSampler)
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