/j5ddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZdZdddZdZd d!dZdZd"d#dZd"d#dZd$dZdS)%) annotations)HUBDatasetStats)Auth)HUBTrainingSession) HUB_API_ROOT HUB_WEB_ROOTPREFIX)LOGGERSETTINGSchecks) rr r check_datasetexport_fmts_hub export_model get_exportloginlogout reset_modelNTapi_key str | Nonesaveboolreturnctjdddlm}td}t jd}|p|}|r |dkrd|ind}||}|jrX|r&|j|krt j d|ji|j|ks|sdnd }tj t|d Stj td |d d S)a\Log in to the Ultralytics HUB API using the provided API key. The session is not stored; a new session is created when needed using the saved SETTINGS or the HUB_API_KEY environment variable if successfully authenticated. Args: api_key (str, optional): API key to use for authentication. If not provided, it will be retrieved from SETTINGS or HUB_API_KEY environment variable. save (bool, optional): Whether to save the API key to SETTINGS if authentication is successful. Returns: (bool): True if authentication is successful, False otherwise. zhub-sdk>=0.0.12r) HUBClientz/settings?tab=api+keysrNu!New authentication successful ✅uAuthenticated ✅TzGet API key from z" and then run 'yolo login API_KEY'F) r check_requirementshub_sdkrrr get authenticatedrupdater infor ) rrr api_key_url saved_key active_key credentialsclient log_messages ]/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/hub/__init__.pyrrs" /000!!!!!!!999K Y''I%IJ-7VJ"W3L3LT_3L / /ex   v,{,,---t  v__ ___```ucRdtd<tjtddS)zJLog out of Ultralytics HUB by removing the API key from the settings file.rru2logged out ✅. To log in again, use 'yolo login'.N)r r r!r r)r(rrDs+HY K6MMMNNNNNr)rmodel_idstrc ddl}|tdd|idtji}|jdkrt jtddSt j td |jd |j dS) z,Reset a trained model to an untrained state.rNz /model-resetmodelId x-api-keyjsonheaderszModel reset successfullyzModel reset failure  ) requestspostrrr status_coder r!r warningreason)r,r6rs r(rrJsOOO 3339h:OZegkgmgmguYv wwA} v777888 NfLL!-LL!(LLMMMMMr)cdddlm}gt|dddddS)z.Return a list of HUB-supported export formats.rexport_formatsArgumentNultralytics_tfliteultralytics_coreml)ultralytics.engine.exporterr>listr=s r(rrUsI:::::: `T.."":.qrr2 3 3 `5I `K_ ``r) torchscriptformatcnddl}|tvsJd|dt|td|dd|idt ji }|jd ks#Jt|d |jd |jtj t|d dS)a>Export a model to a specified format for deployment via the Ultralytics HUB API. Args: model_id (str): The ID of the model to export. An empty string will use the default model. format (str): The format to export the model to. Must be one of the supported formats returned by export_fmts_hub(). Raises: AssertionError: If the specified format is not supported or if the export request fails. Examples: >>> from ultralytics import hub >>> hub.export_model(model_id="your_model_id", format="torchscript") rNUnsupported export format '', valid formats are z /v1/models/z/exportrFr0r1r4z export failure r5u export started ✅) r6rr7rrrr8r r:r r!r,rFr6r;s r(rr\sOOO _&& & & &(vf(v(vcrctct(v(v & & & 55H555Xv>> from ultralytics import hub >>> result = hub.get_export(model_id="your_model_id", format="torchscript") rNrHrIz /get-export)apiKeyr/rFr0r1r4z get_export failure r5) r6rr7rrrr8r r:r2rJs r(rrus"OOO _&& & & &(vf(v(vcrctct(v(v & & & $$$8v N Ndffn-   A =C   F!bF!b!b !b!bXYX`!b!b    6688Or)pathtaskNonect||tjdtddS)aCheck HUB dataset Zip file for errors before upload. Args: path (str): Path to data.zip (with data.yaml inside data.zip). task (str): Dataset task. Options are 'detect', 'segment', 'pose', 'classify', 'obb'. Examples: >>> from ultralytics.hub import check_dataset >>> check_dataset("path/to/coco8.zip", task="detect") # detect dataset >>> check_dataset("path/to/coco8-seg.zip", task="segment") # segment dataset >>> check_dataset("path/to/coco8-pose.zip", task="pose") # pose dataset >>> check_dataset("path/to/dota8.zip", task="obb") # OBB dataset >>> check_dataset("path/to/imagenet10.zip", task="classify") # classification dataset Notes: Download *.zip files from https://github.com/ultralytics/hub/tree/main/example_datasets i.e. https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip for coco8.zip. rMrNu7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rget_jsonr r!rrQs r(r r sE&D)))22444 Kc,cccdddddr))NT)rrrrrr)r)r,r-)rrE)r,r-rFr-)rMr-rNr-rrO) __future__rultralytics.data.utilsrultralytics.hub.authrultralytics.hub.sessionrultralytics.hub.utilsrrr ultralytics.utilsr r r __all__rrrrrrr r+r)r(rZsD#"""""222222%%%%%%666666DDDDDDDDDD6666666666 (((((VOOO NNNNNaaa888882:eeeeeer)