o ci@sddlZddlZddlZddlmmZejGdddZe dkrpe Z e j dde ddd e j d de d d d e \ZZeejejejd ZejZeeegeddZeedeedSdS)Nc@s.eZdZd ddZddZddZd d Zd S) StepActorcCsH||_d|_d|_||_tjjj}|j }t j d||jdddS)NFrJOB:T overwrite) interval_sstopped current_step total_stepsray_privateworker global_worker core_worker get_actor_idray_kv_internal_kv_put)selfrr r worker_idr`/home/ubuntu/.local/lib/python3.10/site-packages/ray/experimental/job/example_job/demo_script.py__init__ s  zStepActor.__init__cCstjjj}|j}|j|jkrD|js9t d|j d|jt |j |jd7_t jd||jddnt dn|j|jksd|_t jd|ddddS) Nz Sleeping z secs to executing step rrTrz"Stop called or reached final step.DONE)r r rrrrr r r printrtimesleeprr)rrrrrrruns(      z StepActor.runcCs|jS)N)r rrrrget_step-szStepActor.get_stepcCs d|_dS)NT)r rrrrstop0s zStepActor.stopN)rr)__name__ __module__ __qualname__rrrr rrrrr s    r__main__z --interval-sFrz Address to use to connect to Ray)requiredtypedefaulthelpz --total-stepsrz Password for connecting to Redis)rr rz, )argparserr ray.experimental.internal_kv experimental internal_kvrremoterr!ArgumentParserparser add_argumentintparse_known_argsargs_initrr step_actorrrefrget_internal_kv_listjob_key_internal_kv_getrrrrs@)