o
    5ti                  	   @   s  d dl Z d dlZd dlZd dlmZ d dlmZ ddlmZ ee	Z
dZdZdZd	Zd
Zg dZejddZed Zed ZdZee Zeedk rTd dlZnd dlmZ h dZedhZeedZ eedZ!ejdd" Z#ejdd" Z$ejdd" Z%dZ&dZ'e$ev re#evre j()dduZ'e'rzeedZ&e
*de& d W n ej+y   Y nw ne
*d dZ,dZ-e#ev re$evre j()dduZ-e-rd D ]Z.z
eee.Z,W n
 ej+y   Y qw  ndZ-e-re,j/d!k re
*d"e, d# dZ-ne
*d$e, d ne
*d% dZ0dZ1e%ev rVe j()d&duZ1e1rUzeed&Z0e
*d'e0 d W n ej+yT   Y nw ne
*d( d)Z2e3d*e2Z4ej56e4d+Z7ej58e3d,e7Z9ej56e9d-Z:ee3d.e:Z;ej56e9d/Z<ee3d0e<Z=ej56e9d1Z>ee3d2e>Z?d3Z@ej56e;e@ZAee3d4eAZBd5ZCej56eAeCZDee3d6eDZEejd7d" ev ZFejd8d" ev ZGd9ZHd:ZId;ZJd<ZKg d=ZLdS )>    N)Path)version   )
get_loggerzAhttps://s3.amazonaws.com/datasets.huggingface.co/datasets/metricsz3https://cdn-datasets.huggingface.co/datasets/metriczWhttps://raw.githubusercontent.com/huggingface/evaluate/{revision}/metrics/{path}/{name}z\https://raw.githubusercontent.com/huggingface/evaluate/{revision}/measurements/{path}/{name}z[https://raw.githubusercontent.com/huggingface/evaluate/{revision}/comparisons/{path}/{name})metric
comparisonmeasurementHF_ENDPOINTzhttps://huggingface.coz/api/spaces?filter={type}z(/spaces/{path}/resolve/{revision}/{name}mainz3.8>   1ONYESTRUEAUTOpandaspyarrowUSE_TF	USE_TORCHUSE_JAXzN/AFtorchzPyTorch version z available.z'Disabling PyTorch because USE_TF is set
tensorflow)	r   ztensorflow-cpuztensorflow-gpuz
tf-nightlyztf-nightly-cpuztf-nightly-gpuzintel-tensorflowztensorflow-rocmztensorflow-macos   z"TensorFlow found but with version z(. `datasets` requires version 2 minimum.zTensorFlow version z-Disabling Tensorflow because USE_TORCH is setjaxzJAX version z-Disabling JAX because USE_JAX is set to Falsez~/.cacheXDG_CACHE_HOMEhuggingfaceHF_HOMEevaluateHF_EVALUATE_CACHEmetricsHF_METRICS_CACHEmodulesHF_MODULES_CACHE	downloads$HF_DATASETS_DOWNLOADED_EVALUATE_PATH	extracted#HF_DATASETS_EXTRACTED_EVALUATE_PATHHF_UPDATE_DOWNLOAD_COUNTSHF_EVALUATE_OFFLINELICENSEzmetric_info.jsonzdataset_dict.jsonevaluate_modules)%zimage-classificationtranslationzimage-segmentationz	fill-maskzautomatic-speech-recognitionztoken-classificationzsentence-similarityzaudio-classificationzquestion-answeringsummarizationzzero-shot-classificationztable-to-textzfeature-extractionotherzmultiple-choiceztext-classificationztext-to-imageztext2text-generationzzero-shot-image-classificationztabular-classificationztabular-regressionzimage-to-imageztabular-to-textzunconditional-image-generationztext-retrievalztext-to-speechzobject-detectionzaudio-to-audioztext-generationconversationalztable-question-answeringzvisual-question-answeringzimage-to-textzreinforcement-learningzvoice-activity-detectionztime-series-forecastingzdocument-question-answering)M	importlibosplatformpathlibr   	packagingr   utils.loggingr   __name__loggerS3_METRICS_BUCKET_PREFIX!CLOUDFRONT_METRICS_DISTRIB_PREFIXREPO_METRICS_URLREPO_MEASUREMENTS_URLREPO_COMPARISONS_URLEVALUATION_MODULE_TYPESenvirongetr	   HF_LIST_ENDPOINTHUB_EVALUATE_URLHUB_DEFAULT_VERSIONparsepython_version
PY_VERSIONimportlib_metadataimportlib.metadatametadataENV_VARS_TRUE_VALUESunionENV_VARS_TRUE_AND_AUTO_VALUESPANDAS_VERSIONPYARROW_VERSIONupperr   r   r   TORCH_VERSIONTORCH_AVAILABLEutil	find_specinfoPackageNotFoundError
TF_VERSIONTF_AVAILABLEpackagemajorJAX_VERSIONJAX_AVAILABLEDEFAULT_XDG_CACHE_HOMEgetenvr   pathjoinDEFAULT_HF_CACHE_HOME
expanduserHF_CACHE_HOMEDEFAULT_HF_EVALUATE_CACHEr   DEFAULT_HF_METRICS_CACHEr   DEFAULT_HF_MODULES_CACHEr!   DOWNLOADED_DATASETS_DIR DEFAULT_DOWNLOADED_EVALUATE_PATHDOWNLOADED_EVALUATE_PATHEXTRACTED_EVALUATE_DIRDEFAULT_EXTRACTED_EVALUATE_PATHEXTRACTED_EVALUATE_PATHr&   r'   LICENSE_FILENAMEMETRIC_INFO_FILENAMEDATASETDICT_JSON_FILENAMEMODULE_NAME_FOR_DYNAMIC_MODULESHF_HUB_ALLOWED_TASKS rn   rn   C/home/ubuntu/.local/lib/python3.10/site-packages/evaluate/config.py<module>   s    




