o
    ॵi                     @   sB   d dl mZ d dlmZmZ edZ	d
dededefdd	ZdS )    )
ConfigDict)Registrybuild_from_cfgcustom_datasetsNcfg	task_namedefault_argsc                 C   s   t | t||dS )aI   Build custom dataset for user-define dataset given model config and task name.

    Args:
        cfg (:obj:`ConfigDict`): config dict for model object.
        task_name (str, optional):  task name, refer to
            :obj:`Tasks` for more details
        default_args (dict, optional): Default initialization arguments.
    )	group_keyr   )r   CUSTOM_DATASETS)r   r   r    r   m/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/msdatasets/dataset_cls/custom_datasets/builder.pybuild_custom_dataset	   s   r   )N)	modelscope.utils.configr   modelscope.utils.registryr   r   r
   strdictr   r   r   r   r   <module>   s   