mirror of https://github.com/THUDM/CodeGeeX.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 lines
304 B
Python
20 lines
304 B
Python
2 years ago
|
from typing import *
|
||
|
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class PromptSample:
|
||
|
|
prompt: str
|
||
|
|
code: str
|
||
|
|
extra: dict = None
|
||
|
|
|
||
|
|
|
||
|
|
PromptDataset = Iterable[PromptSample]
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class LabelSample:
|
||
|
|
prompt: str
|
||
|
|
label: int
|
||
|
|
extra: dict = None
|
||
|
|
|
||
|
|
LabelDataset = Iterable[LabelSample]
|