
                             d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddl	mZ dd	lmZ dd
lmZ ddiZd Z G d dej$                        Zy)z ai-platform jobs update command.    )absolute_import)division)unicode_literals)jobs)base)flags)	jobs_util)labels_util)logEXAMPLESz          To remove all labels in the AI Platform job named ``my-job'', run:

            $ {command} my-job --clear-labels
        c                 l    t         j                  j                  |        t        j                  |        y)z8Get arguments for the `ai-platform jobs update` command.N)r   JOB_NAMEAddToParserr
   AddUpdateLabelsFlagsparsers    &lib/surface/ai_platform/jobs/update.py_AddUpdateArgsr   '   s"    ..V$""6*    c                   *    e Zd ZdZeZed        Zd Zy)UpdatezUpdate an AI Platform job.c                     t        |        y )N)r   r   s    r   ArgszUpdate.Args2   s
    6r   c                     t        j                         }t        j                  ||      }t	        j
                  |j                  d       |S )Nzml engine job)kind)r   
JobsClientr	   r   r   UpdatedResourcejob)selfargsjobs_clientupdated_jobs       r   Runz
Update.Run6   s:    //#K"";5K7r   N)	__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpstaticmethodr   r#    r   r   r   r   -   s!    "- r   r   N)r'   
__future__r   r   r    googlecloudsdk.api_lib.ml_enginer   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer   r	   $googlecloudsdk.command_lib.util.argsr
   googlecloudsdk.corer   r(   r   UpdateCommandr   r+   r   r   <module>r3      sM    ' &  ' 1 ( 6 : < # 	+T r   