
    rh&
                       d dl mZ ddlmZ ddlmZmZ ddlmZm	Z	m
Z
mZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZ dd	lmZmZmZmZm Z  dd
l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ ddl,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4 g dZ5y)    )annotations   )save_to_hub_args_decorator)
all_gatherall_gather_with_grad)check_package_availabilityget_device_nameis_accelerate_availableis_datasets_availableis_training_available)disabled_tqdmhttp_getis_sentence_transformer_modelload_dir_pathload_file_path)mine_hard_negatives)append_to_last_rowdisable_datasets_cachingdisable_loggingfullnameimport_from_string)community_detectioninformation_retrievalparaphrase_miningparaphrase_mining_embeddingssemantic_search)
cos_sim	dot_scoreeuclidean_simmanhattan_simpairwise_angle_simpairwise_cos_simpairwise_dot_scorepairwise_euclidean_simpairwise_manhattan_simpytorch_cos_sim)_convert_to_batch_convert_to_batch_tensor_convert_to_tensorbatch_to_devicenormalize_embeddingsselect_max_active_dimsto_scipy_cootruncate_embeddings)*r   r   r   r	   r   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r   N)6
__future__r   
decoratorsr   distributedr   r   environmentr   r	   r
   r   r   file_ior   r   r   r   r   hard_negativesr   miscr   r   r   r   r   	retrievalr   r   r   r   r   
similarityr   r   r   r    r!   r"   r#   r$   r%   r&   tensorr'   r(   r)   r*   r+   r,   r-   r.   __all__     v/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/sentence_transformers/util/__init__.py<module>r=      sY    " 2 9  k j / m m   	 	 	4r;   