a h @sxddlmZddlZddlZddlZddlmZddddZddd d d Zed d Z eej fddZ ddZ dS)) annotationsN)contextmanagerstr)returncCs8|jj}|dus|tjjkr$|jjS|d|jjSdS)a Gives a full name (package_name.class_name) for a class / object in Python. Will be used to load the correct classes from JSON files Args: o: The object for which to get the full name. Returns: str: The full name of the object. Example: >>> from sentence_transformers.losses import MultipleNegativesRankingLoss >>> from sentence_transformers import SentenceTransformer >>> from sentence_transformers.util import fullname >>> model = SentenceTransformer('all-MiniLM-L6-v2') >>> loss = MultipleNegativesRankingLoss(model) >>> fullname(loss) 'sentence_transformers.losses.MultipleNegativesRankingLoss.MultipleNegativesRankingLoss' N.) __class__ __module__r__name__)omoduler [/var/www/html/assistant/venv/lib/python3.9/site-packages/sentence_transformers/util/misc.pyfullname srtype) dotted_pathrcCsz|dd\}}Wn$ty8|d}t|Yn0zt|}Wntydt|}Yn0z t||WStyd|d|d}t|Yn0dS)a: Import a dotted module path and return the attribute/class designated by the last name in the path. Raise ImportError if the import failed. Args: dotted_path (str): The dotted module path. Returns: Any: The attribute/class designated by the last name in the path. Raises: ImportError: If the import failed. Example: >>> import_from_string('sentence_transformers.losses.MultipleNegativesRankingLoss') rz doesn't look like a module pathzModule "z" does not define a "z" attribute/classN)rsplit ValueError ImportError importlib import_module ExceptiongetattrAttributeError)r module_path class_namemsgr r r r import_from_string%s     rccsJddlm}m}m}|}z|r&|dVW|rF|n |rD|0dS)zN A context manager that will disable caching in the datasets library. r)disable_cachingenable_cachingis_caching_enabledN)Zdatasetsrrr )rrr Zis_originally_enabledr r r disable_datasets_cachingIsr!c cs:tjjj}t|zdVWt|n t|0dS)z A context manager that will prevent any logging messages triggered during the body from being processed. Args: highest_level: the maximum logging level allowed. N)loggingrootmanagerdisable)Z highest_levelZprevious_levelr r r disable_logging[s  r&cCst|ddd"}t|}t|}Wdn1s60Yt|dkr|d|t|dddd$}t|}||Wdn1s0YdSdS) Nzutf-8)newlineencodingrwTF)opencsvreaderlistlenextendwriter writerows)Zcsv_pathZadditional_datafr.rowsr2r r r append_to_last_rowms &  (r6) __future__rr-rr" contextlibrrrr!CRITICALr&r6r r r r s  $