mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-27 16:10:30 +08:00
Yada Pruksachatkun
d1d185edb6
* Adding clinical note dataset * Fix flake8 issues * Fix prepare.py for straggling commas, replace assistant with Rosey in prompt Co-authored-by: Yada P <yadapruk@Tests-MacBook-Pro.local>
22 lines
399 B
Python
22 lines
399 B
Python
from dataclasses import dataclass
|
|
|
|
import datasets
|
|
|
|
|
|
@dataclass
|
|
class OpenAssistantConfig(datasets.BuilderConfig):
|
|
"""BuilderConfig for OpenAssistant datasets."""
|
|
|
|
name: str = None
|
|
version: datasets.Version = None
|
|
description: str = None
|
|
schema: str = None
|
|
subset_id: str = None
|
|
|
|
|
|
features = datasets.Features(
|
|
{
|
|
"conversation": datasets.Value("string"),
|
|
}
|
|
)
|