b vdZddlmZddlmZddlmZddlmZddlmZddl m Z dZ d d d d Z Gd de Zy)z2Utilities for Edge ML API long-running operations.)absolute_import)division)unicode_literals)util)waiter) resourcesi' z+Analyzing Edge machine learning model. [{}]z9Optimizing Edge machine learning model for Edge TPU. [{}]z,Converting SavedModel to TF Lite model. [{}])analyzecompileconvertc$eZdZdZddZdZdZy)OperationsClientzClient for the Edge ML operations API. Attributes: client: Generated Edge ML API client. messages: Generated Edge ML API messages. Nc|xstj|_|xstj||_|jj |_y)N)rGetClientInstanceclientGetMessagesModulemessages operations_service)selfrrs /lib/googlecloudsdk/api_lib/edgeml/operations.py__init__zOperationsClient.__init__+s?4D224DK> 6 6v >DMKK**DMc|jjj|jj |j S)zCalls get method for long-running operations. Args: operation_ref: Ref to long-running operation. Returns: edgeml.Operation message. )name)rprojects_operationsGetrEdgemlOperationsGetRequest RelativeName)r operation_refs rrzOperationsClient.Get0sD ;; * * . . 00++- 1 / 00rc Jtj|j}tjj |j d}|jjdd}tj|d}tj|||dtdddd S) aJWait until the operation is complete or times out. Args: operation: The operation resource to wait on. Returns: The operation resource when it has completed. Raises: OperationTimeoutError: When the operation polling times out. OperationError: When the operation completed with an error. zedgeml.operations) collection/rzWaiting for operation [{}]iNi)pre_start_sleep_ms max_wait_msexponential_sleep_multiplier jitter_mswait_ceiling_mssleep_ms) rCloudOperationPollerNoResourcesrrREGISTRYParserNamesplit_WAITING_MESSAGEgetWaitFor_MAX_WAIT_TIME_MS)r operationpollerroperation_typemessages rWaitForOperationz!OperationsClient.WaitForOperation=s 3 3DMM BF&&,,#6-8M"'')//4Q7N"">3OPG >>%%)  r)NN)__name__ __module__ __qualname____doc__rrr6rrr r #s+ 0rr N)r: __future__rrrgooglecloudsdk.api_lib.edgemlrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corerr1r.objectr r;rrrAsC9&'..)#=J=5v5r