o ϖi&5@sddlmZmZmZmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZmZddlmZddlmZddlmZddlmZmZddlmZzdd lmZWn ey]dd lZYnwerldd lm Z dd l!m"Z"Gd ddeZ#GdddeZ$ d(d d d d ddeee%efdee e%ee%e%fdfdee%dee%deee%dee%eff ddZ&de%de%fddZ'de e%ee%e%fdfde e%d fde%fd!d"Z(d#d$d%ee%efde#fd&d'Z)d S))) TYPE_CHECKINGAnyDictListLiteralOptionalTuple TypedDictUnionoverload) NotRequiredUnpack)async_encode_json encode_json)ModelVersionIdentifier)Model)Page) NamespaceResource)Version)v1N)Client)FileEncodingStrategyc@s eZdZUdZeZded<eed< eed< e ee fed< e eed< e ded < e e eefed < e eed < e eed < e eed < e eed< e eed< e eed< e e eefed< dddZdddZdddZdddZdS)Trainingz: A training made for a model hosted on Replicate. r_clientidmodelversion destination)starting processing succeededfailedcanceledstatusinputoutputlogserror created_at started_at completed_aturlsreturnNcC6|jj|j}|D] \}}t|||qdS)z, Cancel a running training. N)r trainingscancelrdictitemssetattrselfr#namevaluer8F/home/ubuntu/.local/lib/python3.10/site-packages/replicate/training.pyr0UzTraining.cancelc>|jj|jIdH}|D] \}}t|||qdS)z; Cancel a running training asynchronously. N)rr/ async_cancelrr1r2r3r4r8r8r9r<^ zTraining.async_cancelcCr.)z4 Load the training from the server. N)rr/getrr1r2r3r5updatedr6r7r8r8r9reloadgr:zTraining.reloadcr;)zC Load the training from the server asynchronously. N)rr/ async_getrr1r2r3r?r8r8r9 async_reloadpr=zTraining.async_reload)r-N)__name__ __module__ __qualname____doc__pydantic PrivateAttrr__annotations__strr rrrrrr0r<rArCr8r8r8r9r!sB            rc@seZdZdZd&deeddfdeefddZ d&deeddfdeefd d Z d edefd d Z d edefddZ Gddde Z e  d'dedeeefdedeedeeedef ddZe d(deeeeefdfdeeefdeeeefdeddef ddZdddddeeeeeefdfdeeeefdeeeefdeddef ddZdeeeeefdfdeeefdeeefdeddef d d!Zd edefd"d#Zd edefd$d%ZdS)) Trainingsz8 Namespace for operations related to trainings. .cursorellipsisNr-cs^|durtdjd|durdn|}|}fdd|dD|d<ttd i|S) # List your trainings. Parameters: cursor: The cursor to use for pagination. Use the value of `Page.next` or `Page.previous`. Returns: Page[Training]: A page of trainings. Raises: ValueError: If `cursor` is `None`. Ncursor cannot be NoneGET. /v1/trainingscg|]}tj|qSr8_json_to_trainingr.0resultr5r8r9  z"Trainings.list..resultsr8) ValueErrorr_requestjsonrrr5rMrespobjr8rYr9lists   zTrainings.listcsf|dur tdjd|durdn|IdH}|}fdd|dD|d<ttd i|S) rONrPrQ.rRcrSr8rTrVrYr8r9rZr[z(Trainings.async_list..r\r8)r]r_async_requestr_rrr`r8rYr9 async_lists   zTrainings.async_listrcCs$|jdd|}t|j|S) Get a training by ID. Args: id: The ID of the training. Returns: Training: The training object. rQ/v1/trainings/rr^rUr_r5rrar8r8r9r>s  z Trainings.getcs,|jdd|IdH}t|j|S)rfrQrgNrrdrUr_rir8r8r9rBs   zTrainings.async_getc@sheZdZUdZeeeeefdfed< eeed< eeed< ee eed< eded<d S) Trainings.CreateTrainingParamsz#Parameters for creating a training.rrwebhookwebhook_completedwebhook_events_filterrfile_encoding_strategyN) rDrErFrGr rKrrJr rr8r8r8r9CreateTrainingParamss    rprr%rrlrncKdSNr8)r5rr%rrlrnkwargsr8r8r9creates zTrainings.createrrparamsrkcKrqrrr8)r5rrr%rur8r8r9rts)rrr%c Os6d}|rSt|dkr|dnd}rt|}t|dkr |dn|}t|dkr.|d|d<t|dkr:|d|d<t|dkrF|d|d <t|d krR|d |d <n|r]|r]t||}n |durjt|trjt|}|sptd |d d}|durt||j|d}t |fi|} |jj d|| d} t |j| S)zT Create a new training using the specified model version as a base. Nrrrlrmrnz8model and version or shorthand version must be specifiedroclientroPOSTr_) len#_create_training_url_from_shorthand+_create_training_url_from_model_and_version isinstancerKr]poprr_create_training_bodyr^rUr_) r5rrr%argsruurl shorthandrobodyrar8r8r9rtsD           c snt||}|dd}|durt||j|dIdH}t|fi|}|jjd||dIdH}t|j|S)a Create a new training using the specified model version as a base. Args: version: The ID of the base model version that you're using to train a new model version. input: The input to the training. destination: The desired model to push to in the format `{owner}/{model_name}`. This should be an existing model owned by the user or organization making the API request. webhook: The URL to send a POST request to when the training is completed. Defaults to None. webhook_completed: The URL to receive a POST request when the prediction is completed. webhook_events_filter: The events to send to the webhook. Defaults to None. Returns: The training object. roNr{r}r~)rrrrrrdrUr_) r5rrr%rurrorrar8r8r9 async_create4s     zTrainings.async_createcCs&|jdd|d}t|j|S) Cancel a training. Args: id: The ID of the training to cancel. Returns: Training: The canceled training object. r}rg/cancelrhrir8r8r9r0\s   zTrainings.cancelcs.|jdd|dIdH}t|j|S)rr}rgrNrjrir8r8r9r<ms    zTrainings.async_cancel).)NNrr)rDrErFrGr rKrrrcrer>rBr rpr rrrrrtrrr rr0r<r8r8r8r9rLzs"          6   (rL)rrlrmrnr%rrrlrmrnr-cKsi}|dur ||d<|durtdt|tr!|jd|j}nt|tr1|dd|d}||d<|dur=||d<|durE||d<|durM||d <|S) Nr%zCA destination must be provided as a positional or keyword argument./rrvrrlrmrn)r]rrownerr6tuple)r%rrlrmrn_kwargsrr8r8r9rs&   rrefcCs(t|\}}}d|d|d|dS)N /v1/models/r /versions/ /trainings)rparse)rrr6 version_idr8r8r9rsrrrrcCst|tr |j|j}}n!t|tr|d|d}}nt|tr*t|\}}}ntdt|t r7|j }n|}d|d|d|dS)NrrvzWmodel must be a Model, a tuple of (owner, name), or a string in the format 'owner/name'rrrr) rrrr6rrKrrr]rr)rrrr6_rr8r8r9rs    rr|rr_cCsXtdi|}||_|jr*t|jtr*d|jvr*t|jd}|jd|j|_ |S)Nrrr8) rrr&rr1rrrr6r)r|r_trainingrr8r8r9rUs  rUrr)*typingrrrrrrrr r r typing_extensionsr r replicate.helpersrrreplicate.identifierrreplicate.modelrreplicate.paginationrreplicate.resourcerrreplicate.versionrrHr ImportErrorreplicate.clientrreplicate.filerrrLrKrrrrUr8r8r8r9s`0         Y   $  "