SrSSKJrJrJr SSKJr SSKJr SSKJ r S\RS\ RRS\\ S \\4S jrS\RS\ RR S \4S jrSS\RS\ RR S\\S\\ S\\\\4S\\\S\\S\\4SjjrSS\RS\ RR S\4Sjjrg )z&The BigQuery CLI model client library.)DictListOptional) discovery)bq_error) bq_id_utils model_client reference max_results page_tokencUR5RURURUUS9R 5$)aLists models for the given dataset reference. Arguments: model_client: The apiclient used to make the request. reference: Reference to the dataset. max_results: Number of results to return. page_token: Token to retrieve the next page of results. Returns: A dict that contains entries: 'results': a list of models 'token': nextPageToken for the last page, if present. ) projectId datasetId maxResults pageToken)modelslistrrexecute)r r r r s #platform/bq/clients/client_model.py list_modelsr sE( t''''    wy returnc*[R"U[RRSS9 UR 5R UR URURS9R5$![Ra gf=f)z!Returns true if the model exists. model_existsmethodrrmodelIdF) r typecheckApiClientHelperModelReferencergetrrrrrBigqueryNotFoundError)r r s rrr+s !!00   ))))%%    ' ' sA A;;BBN description expiration labels_to_setlabel_keys_to_removevertex_ai_model_idetagc[R"U[RRSS9 0nUbX(S'UbU=(d SUS'SU;a0US'U(a UR 5H upXSU 'M U(aUH n SUSU 'M UbSU0/US'UR 5R UURURURS 9n U(aU(aUOUS U RS 'U R5 g) aUpdates a Model. Args: model_client: The apiclient used to make the request. reference: the ModelReference to update. description: an optional description for model. expiration: optional expiration time in milliseconds since the epoch. Specifying 0 clears the expiration time for the model. labels_to_set: an optional dict of labels to set on this model. label_keys_to_remove: an optional list of label keys to remove from this model. vertex_ai_model_id: an optional string as Vertex AI model ID to register. etag: if set, checks that etag in the existing model matches. Raises: BigqueryTypeError: if reference is not a ModelReference. update_modelrNr$expirationTimelabelsr( trainingRuns)bodyrrrr)zIf-Match) rrr r!itemsrpatchrrrheadersr) r r r$r%r&r'r(r) updated_model label_key label_valuerequests rr+r+Cs!6!!00  -#.- &0&8DM"# ]" M("/"5"5"7 +6Hi(#8) +/mHi(*#&:rEs_,''%$$**;;# >$$**99 6"& $.204(,>$$>**99>#> > DcN+ > #49- >! > 3->H# $$ **99  r