o
    "i                     @   sp   d Z ddlmZmZmZmZmZ ddlmZ ddl	m
Z
 ddlmZ eG dd de
ZeG dd	 d	e
Zd
S )z
Training Dataset models
    )ColumnIntegerStringText
ForeignKey)JSONB)	BaseModel)global_modelc                   @   s   e Zd ZdZdZddiZeedddZee	ddZ
eed	ddZeed
dZeeddZeeddZeeddZeeddZeeddZeeddZeei dZdS )TrainingDatasetzTraining dataset modeltraining_datasetsschema
hub_global   FnullableT2      defaultr   N)__name__
__module____qualname____doc____tablename____table_args__r   r   namer   descriptiondataset_typer   versionparent_version_idtotal_samplestrain_samplesvalidation_samplestest_samples
created_byr   	meta_data r&   r&   M/lsinfo/ai/hellotax_ai/training_center/backend/app/models/training_dataset.pyr
   
   s    r
   c                   @   sr   e Zd ZdZdZddiZeeedddZ	ee
ddZeedddZeed	d
dZeeddZeei dZdS )DatasetSamplezDataset sample modeldataset_samplesr   r   zhub_global.training_datasets.idFr   r   r   trainr   TN)r   r   r   r   r   r   r   r   r   
dataset_idr   contentr   labelsplitsource_task_idr   r%   r&   r&   r&   r'   r(      s    r(   N)r   
sqlalchemyr   r   r   r   r   sqlalchemy.dialects.postgresqlr   app.models.baser   app.core.schema_bindingr	   r
   r(   r&   r&   r&   r'   <module>   s    