ai dZddlZddlZddlmZmZmZddlZddlm Z m Z ddl m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZeee Zee Zej<d eDcic]}||c}e Z eeeejBd d jEed eefZ#edZ$e$jKd d!dedededeee ejBdfdede#deddfdZ&e$jKd d"deeejNdfdede#deddf d Z(ycc}w)#aContains commands to interact with datasets on the Hugging Face Hub. Usage: # list datasets on the Hub hf datasets ls # list datasets with a search query hf datasets ls --search "code" # get info about a dataset hf datasets info HuggingFaceFW/fineweb N) AnnotatedOptionalget_args)RepositoryNotFoundErrorRevisionNotFoundError) DatasetSort_TExpandDatasetProperty_T)ANSI) AuthorOpt FilterOptLimitOpt RevisionOpt SearchOptTokenOpt get_hf_apimake_expand_properties_parserrepo_info_to_dict typer_factoryDatasetSortEnum)typezWComma-separated properties to expand. Example: '--expand=downloads,likes,tags'. Valid: z, .)helpcallbackz"Interact with datasets on the Hub.)rlssearchauthorfiltersortz Sort results.limitexpandtokenreturnct|}|r |jnd}|j||||||D cgc] } t| } } t t j | dycc} w)zList datasets on the Hub.r"N)rrrrr r!indent)rvalue list_datasetsrprintjsondumps) rrrrr r!r"apisort_key dataset_inforesultss f/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/huggingface_hub/cli/datasets.py datasets_lsr3Dsw 5 !C!tzztH --&he\b.   ,'G  $**WQ '( sA+info dataset_idz+The dataset ID (e.g. `username/repo-name`).revisionc t|} |j|||}ttjt|d y #t$r9tdt j |dt jdt$rXtdt j t|dt j |d t jdwxYw) z$Get info about a dataset on the Hub.r%)repo_idr6r!zDataset z not found.r )codez Revision z not found on rr&r'N) rr0rr+r boldtyperExitrstrr,r-r)r5r6r!r"r.r4s r2 datasets_infor>]s 5 !C! XfU $**&t,Q 78 #! :./{;<jja  ! $))CM23>$))JBWAXXYZ[jja  !s A B!C-)NNNN NN)NNN))__doc__enumr,typingrrrr;huggingface_hub.errorsrrhuggingface_hub.hf_apirr huggingface_hub.utilsr _cli_utilsr r rrrrrrrrsorted_EXPAND_PROPERTIES _SORT_OPTIONSEnumr=rOptionjoin ExpandOpt datasets_clicommandr3Argumentr>)ss0r2rRs  00 QI&   H%<=>' $))-m/L1/LSVW  SMELLfgkgpgprDhEgFFGH./AB  "FG d ) ) ) ) ! /* ,  ) ) ) ) ))0f! 9#~u~~3`aab99 9  9  99S0Ms. E