dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZe jGd d e j Zy ) Submit a Ray batch job.)absolute_import)division)unicode_literals)dataproc)base)flags)batch_submitter)ray_batch_factoryc.eZdZdZddiZedZdZy)RayrEXAMPLESz 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 cXtj|tj|yN)r AddArgumentsr AddLocationFlag)parsers *lib/surface/dataproc/batches/submit/ray.pyArgszRay.Args)s""6* &!ctjtjj}t j |j|}tj|||Sr) dpDataprocr ReleaseTrackBETAr RayBatchFactoryUploadLocalFilesAndGetMessager Submit)selfargsr ray_batchs rRunzRay.Run.sT{{4,,112H!11//5  ! !)Xt <r1sE &':(5GI=$,,= =r