o `۷i@sdddlmZmZddlmZmZmZmZmZm Z ddl m Z GdddeZ GdddZ e ZdS) )Enumauto)AnyCallableDictListOptionalUnion) cli_loggerc@sReZdZdZeZeZeZeZeZ eZ eZ eZ eZ eZeZdS)CreateClusterEventaEvents to track in ray.autoscaler.sdk.create_or_update_cluster. Attributes: up_started : Invoked at the beginning of create_or_update_cluster. ssh_keypair_downloaded : Invoked when the ssh keypair is downloaded. cluster_booting_started : Invoked when when the cluster booting starts. acquiring_new_head_node : Invoked before the head node is acquired. head_node_acquired : Invoked after the head node is acquired. ssh_control_acquired : Invoked when the node is being updated. run_initialization_cmd : Invoked before all initialization commands are called and again before each initialization command. run_setup_cmd : Invoked before all setup commands are called and again before each setup command. start_ray_runtime : Invoked before ray start commands are run. start_ray_runtime_completed : Invoked after ray start commands are run. cluster_booting_completed : Invoked after cluster booting is completed. N)__name__ __module__ __qualname____doc__r up_startedssh_keypair_downloadedcluster_booting_startedacquiring_new_head_nodehead_node_acquiredssh_control_acquiredrun_initialization_cmd run_setup_cmdstart_ray_runtimestart_ray_runtime_completedcluster_booting_completedrrZ/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/autoscaler/_private/event_system.pyr s r c @szeZdZdZddZdedeeegdfe eegdfffddZ dde d e eee ffd d Zdefd d ZdS) _EventSystemzEvent system that handles storing and calling callbacks for events. Attributes: callback_map (Dict[str, List[Callable]]) : Stores list of callbacks for events when registered. cCs i|_dSN callback_map)selfrrr__init__1s z_EventSystem.__init__eventcallbackNcCsN|tjvrt|d|j|gt|t ur"|gdS|dS)agStores callback handler for event. Args: event: Event that callback should be called on. See CreateClusterEvent for details on the events available to be registered against. callback (Callable[[Dict], None]): Callable object that is invoked when specified event occurs. zA is not currently tracked, and this callback will not be invoked.N) r __members__valuesr warningr setdefaultextendtypelist)r!r#r$rrradd_callback_handler4sz!_EventSystem.add_callback_handler event_datacCs>|duri}||d<||jvr|j|D]}||qdSdS)a:Executes all callbacks for event. Args: event: Event that is invoked. See CreateClusterEvent for details on the available events. event_data (Dict[str, Any]): Argument that is passed to each callable object stored for this particular event. N event_namer)r!r#r-r$rrrexecute_callbackLs   z_EventSystem.execute_callbackcCs||jvr |j|=dSdS)zClears stored callable objects for event. Args: event: Event that has callable objects stored in map. See CreateClusterEvent for details on the available events. Nr)r!r#rrrclear_callbacks_for_event_s  z&_EventSystem.clear_callbacks_for_eventr)r r rrr"strr rrrr,r rrr/r0rrrrr)s"  rN)enumrrtypingrrrrrr "ray.autoscaler._private.cli_loggerr r rglobal_event_systemrrrrs   " A