
    g                         S r SSKJr  SSKJr  SSKJr  SSKJ	r	  SSKJ
r
  \R                   " S S\R                  5      5       rg	)
Submit a Ray batch job.    )dataproc)base)flags)batch_submitter)ray_batch_factoryc                   6    \ rS rSrSrSS0r\S 5       rS rSr	g)	Ray   r   EXAMPLESz          To submit a Ray batch job called "my-batch" that runs "my-ray.py", run:

            $ {command} my-ray.py --batch=my-batch --deps-bucket=gs://my-bucket --location=us-central1
          c                 \    [         R                  " U 5        [        R                  " U 5        g N)r   AddArgumentsr   AddLocationFlag)parsers    *lib/surface/dataproc/batches/submit/ray.pyArgsRay.Args&   s    ""6*	&!    c                     [         R                  " [        R                  R                  5      n[
        R                  " U5      R                  U5      n[        R                  " X2U5      $ r   )
dpDataprocr   ReleaseTrackBETAr   RayBatchFactoryUploadLocalFilesAndGetMessager   Submit)selfargsr   	ray_batchs       r   RunRay.Run+   sR    {{4,,112H!11//5  !!)t<<r    N)
__name__
__module____qualname____firstlineno____doc__detailed_helpstaticmethodr   r!   __static_attributes__r#   r   r   r
   r
      s-    - " "=r   r
   N)r(   googlecloudsdk.api_lib.dataprocr   r   googlecloudsdk.callioper   #googlecloudsdk.command_lib.dataprocr   +googlecloudsdk.command_lib.dataproc.batchesr   r   HiddenCommandr
   r#   r   r   <module>r2      s>      ; ( 5 G I =$,, = =r   