
    h^
                       S SK Jr  SSKJr  SSKJrJr  SSKJrJ	r	J
r
JrJr  SSKJrJrJrJrJr  SSKJr  SSKJrJrJrJrJr  SS	KJrJrJrJrJ r   SS
K!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+  SSK,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5  / SQr6g)    )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compute_count_vector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+   r   N)7
__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.   r/   __all__     ]/home/james-whalen/.local/lib/python3.13/site-packages/sentence_transformers/util/__init__.py<module>r>      sY    " 2 9  k j / m m   
 
 
5r<   