o
    à¥µit  ã                   @   sZ   d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ G dd„ dƒZdS )	é    )ÚMappingÚSequenceÚUnion)ÚBaseAuthConfig)ÚDataDownloadConfig)ÚDataMetaConfig)ÚDownloadModeÚHubsc                   @   sŠ  e Zd ZdZdeeef dedededeeef deded	ed
eeee e	eeeee f f f de
dededefdd„Zedefdd„ƒZejdefdd„ƒZedefdd„ƒZejdefdd„ƒZedefdd„ƒZejdefdd„ƒZedefdd„ƒZejdefdd„ƒZedefdd „ƒZejdefd!d „ƒZedefd"d#„ƒZejdefd$d#„ƒZedefd%d&„ƒZejdefd'd&„ƒZd(S ))ÚDatasetContextConfigz!Context configuration of dataset.Údataset_nameÚ	namespaceÚversionÚsubset_nameÚsplitÚtargetÚhubÚdata_dirÚ
data_filesÚdownload_modeÚcache_root_dirÚuse_streamingÚstream_batch_sizec                 K   s‚   d | _ d | _|| _d | _d | _d | _d | _|| _|
| _|| _	|| _
|| _|| _|| _|| _|| _|	| _|| _|| _|| _d| _d S )NF)Ú_download_configÚ_data_meta_configÚ_config_kwargsÚ_dataset_version_cache_root_dirÚ_auth_configÚ_global_meta_lock_file_pathÚ_global_data_lock_file_pathr   r   r   r   r   r   r   r   r   r   r   r   r   Údownload_virgo_files)Úselfr   r   r   r   r   r   r   r   r   r   r   r   r   Úkwargs© r"   úh/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/msdatasets/context/dataset_context_config.pyÚ__init__   s*   
zDatasetContextConfig.__init__Úreturnc                 C   ó   | j S ©N©r   ©r    r"   r"   r#   Úconfig_kwargs0   ó   z"DatasetContextConfig.config_kwargsÚvalc                 C   ó
   || _ d S r'   r(   ©r    r,   r"   r"   r#   r*   4   ó   
c                 C   r&   r'   ©r   r)   r"   r"   r#   Údownload_config8   r+   z$DatasetContextConfig.download_configc                 C   r-   r'   r0   r.   r"   r"   r#   r1   <   r/   c                 C   r&   r'   ©r   r)   r"   r"   r#   Údata_meta_config@   r+   z%DatasetContextConfig.data_meta_configc                 C   r-   r'   r2   r.   r"   r"   r#   r3   D   r/   c                 C   r&   r'   ©r   r)   r"   r"   r#   Údataset_version_cache_root_dirH   r+   z3DatasetContextConfig.dataset_version_cache_root_dirc                 C   r-   r'   r4   r.   r"   r"   r#   r5   L   r/   c                 C   r&   r'   ©r   r)   r"   r"   r#   Úglobal_meta_lock_file_pathP   r+   z/DatasetContextConfig.global_meta_lock_file_pathc                 C   r-   r'   r6   r.   r"   r"   r#   r7   T   r/   c                 C   r&   r'   ©r   r)   r"   r"   r#   Úglobal_data_lock_file_pathX   r+   z/DatasetContextConfig.global_data_lock_file_pathc                 C   r-   r'   r8   r.   r"   r"   r#   r9   \   r/   c                 C   r&   r'   ©r   r)   r"   r"   r#   Úauth_config`   r+   z DatasetContextConfig.auth_configc                 C   r-   r'   r:   r.   r"   r"   r#   r;   d   r/   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚstrÚlistr	   r   r   r   ÚboolÚintr$   ÚpropertyÚdictr*   Úsetterr   r1   r   r3   r5   r7   r9   r   r;   r"   r"   r"   r#   r
      sn    ÿÿ
ÿþþþ
ÿýûûú
ú"r
   N)Útypingr   r   r   Ú&modelscope.msdatasets.auth.auth_configr   Ú.modelscope.msdatasets.download.download_configr   Ú+modelscope.msdatasets.meta.data_meta_configr   Úmodelscope.utils.constantr   r	   r
   r"   r"   r"   r#   Ú<module>   s   