^j0dZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZmZdd lmZmZd Zej.eZdd ZGd dZy#e$rd ZY*wxYw)zThis module provides mixin functionality for the Albumentations library. It includes utility functions and classes to enhance the core capabilities. ) annotationsN)Path)AnyCallable)load)save)HfApihf_hub_download)HfHubHTTPErrorSoftTemporaryDirectoryTFcDtjdfd }|S)aDecorator to require huggingface_hub. This decorator ensures that the `huggingface_hub` package is installed before executing the decorated function. If the package is not installed, it raises an ImportError with instructions on how to install it. Args: func (Callable[..., Any]): The function to decorate. Returns: Callable[..., Any]: The decorated function. cRtstdjd|i|S)Nz-You need to install `huggingface_hub` to use zJ. Run `pip install huggingface_hub`, or `pip install albumentations[hub]`.)is_huggingface_hub_available ImportError__name__)argskwargsfuncs h/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/albumentations/core/hub_mixin.pywrapperz(require_huggingface_hub..wrapper)s=+? O[[ T$V$$)rrrrreturnr) functoolswraps)rrs` rrequire_huggingface_hubrs'__T%% Nrc eZdZdZdZdZddZeddZddddd  dd Z edddddddd  dd Z e ddd ddddd ddZ y)HubMixinaMixin class for Hugging Face Hub integration. This class provides functionality for saving and loading transforms to/from the Hugging Face Hub. It enables serialization, deserialization, and sharing of transform configurations. Args: _CONFIG_KEYS (tuple[str, ...]): Keys used for configuration files. _CONFIG_FILE_NAME_TEMPLATE (str): Template for configuration filenames. )trainevalzalbumentations_config_{}.jsoncht|}|jdd||z }t||d|S)aKSave the transform to a specified directory. Args: save_directory (Union[str, Path]): Directory where the transform will be saved. filename (str): Name of the file to save the transform. Returns: Path: Path to the saved transform file. T)parentsexist_okjson data_format)rmkdirsave_transform)selfsave_directoryfilename save_paths r_save_pretrainedzHubMixin._save_pretrainedEs?n-TD9"X-  tYF;rc8t||z }t|dS)aKLoad a transform from a specified directory. Args: save_directory (Union[str, Path]): Directory from where the transform will be loaded. filename (str): Name of the file to load the transform from. Returns: A.Compose: Loaded transform. r#r$)rload_transform)clsr)r*r+s r_from_pretrainedzHubMixin._from_pretrained\s (83 iV<!S0A0A%A % nh7 ',,.F~.33#4##GGGG Gr)r1force_downloadproxiestoken cache_dirlocal_files_onlyrevisionc |jj|} t|}d} |jr| |j D cgc]} | j c} vr|j || } natr3tj| dt|jdn(t| dt|jd| | S tt|jdd| ||||||} t| jt| j } } |j | | Scc} w#t $r}t!| d |d}~wwxYw) aLoad a transform from the Huggingface Hub or a local directory. Args: directory_or_repo_id (`str`, `Path`): - Either the `repo_id` (string) of a repo with hosted transform on the Hub, e.g. `qubvel-hf/albu`. - Or a path to a `directory` containing transform config saved using [`~albumentations.Compose.save_pretrained`], e.g., `../path/to/my_directory/`. key (`str`, *optional*): Key to identify the configuration type, one of ["train", "eval"]. Defaults to "eval". revision (`str`, *optional*): Revision of the repo on the Hub. Can be a branch name, a git tag or any commit id. Defaults to the latest commit on `main` branch. force_download (`bool`, *optional*, defaults to `False`): Whether to force (re-)downloading the transform configuration files from the Hub, overriding the existing cache. proxies (`dict[str, str]`, *optional*): A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}`. The proxies are used on every request. token (`str` or `bool`, *optional*): The token to use as HTTP bearer authorization for remote files. By default, it will use the token cached when running `huggingface-cli login`. cache_dir (`str`, `Path`, *optional*): Path to the folder where cached files are stored. local_files_only (`bool`, *optional*, defaults to `False`): If `True`, avoid downloading the file and return the path to the local cached file if it exists. N)r)r*z not found in z, trying to load from the Hub.z8. Please install `huggingface_hub` to load from the Hub.\/)r3r*rFrDrArBrCrEz! not found on the HuggingFace Hub)r;r<ris_diriterdirr>r0rloggerinforesolveFileNotFoundErrorr strreplaceparentr )r/directory_or_repo_idr1rArBrCrDrErFr* transformf config_file directoryes rfrom_pretrainedzHubMixin.from_pretrainedsP1188=#$89  & & (,@,H,H,JKqAFFKK00@T_g0h - jt4H/I/Q/Q/S.TTrs(jt4H/I/Q/Q/S.TUNN    X)0199$D!!#-!1 K#'{"3":":D>C)r) str | Pathr*rPrr)r)rer*rProbject)r)rer1rPr2boolr3 str | Noner4rgr?rrrh)r/rrSrer1rPrArgrBzdict[str, str] | NonerCzstr | bool | NonerDzstr | Path | NonerErgrFrhrrf)r3rPr1rPr2rgrZrPr[rgrCrhr\rhr]z bool | NonerrP) r __module__ __qualname____doc__r9r;r, classmethodr0r@rYrr4r8rrrr5s %L!@.==("'"!/"/ /  /  //"/ /b $)-#''+!&#KX KX(KX KX  KX ' KX!KX%KXKXKX KXKXZ "'E !!%>> >  >  >>>>> >>rr)rCallable[..., Any]rrm)rk __future__rrloggingpathlibrtypingrr!albumentations.core.serializationrr.rr'huggingface_hubr r huggingface_hub.utilsr r rr getLoggerrrLrrr8rrrvso# DD)6L#'    8 $6vvC)#( )sAA)(A)