o $i@sddlZddlZddlmZmZddlmZddlmZdZ dZ dee efdee effd d Z dee efdee effd d Z dee efdee effd dZde de fddZde de fddZdee efdee effddZdS)N)AnyDict)get_ray_temp_dir) cli_loggerz6The field {} is not supported for on-premise clusters.zlocal.cluster.nodeconfigreturncCspt|}dD]}||rt|}t|qtiidi|d<t|d<d|dvr2t|}|St |}|S)z\ Prepare local cluster config for ingestion by cluster launcher and autoscaler. ) head_node worker_nodesavailable_node_types) node_config resourcesr head_node_typecoordinator_addressprovider) copydeepcopygetunsupported_field_messageformatrabortLOCAL_CLUSTER_NODE_TYPEprepare_coordinatorprepare_manual)rfielderr_msgra/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/autoscaler/_private/local/config.py prepare_local s      rcCsHt|}d|vrtd|dt}|dd|d<|d|d<|S)N max_workersz[The field `max_workers` is required when using an automatically managed on-premise cluster.r min_workersr)rrrrrpop)r node_typerrrr%s   rc Cs t|}d|dvsd|dvrtdt|dd}|dt}|d||d|}|d}||krLtd|d |d |d ||d<n||d<||krltd |d |d |d ||d<||d<n||d<||krtd |d|d|d|S)ahValidates and sets defaults for configs of manually managed on-prem clusters. - Checks for presence of required `worker_ips` and `head_ips` fields. - Defaults min and max workers to the number of `worker_ips`. - Caps min and max workers at the number of `worker_ips`. - Writes min and max worker info into the single worker node type. worker_ipsrhead_ipzbPlease supply a `head_ip` and list of `worker_ips`. Alternatively, supply a `coordinator_address`.r rrz%The value of `min_workers` supplied (z6) is greater than the number of available worker ips (z). Setting `min_workers=z`.z%The value of `max_workers` supplied (z). Setting `max_workers=z3) is less than the number of available worker ips (z ). At most z. Ray worker nodes will connect to the cluster.) rrrrlenr setdefaultr warning)rnum_ipsr!rrrrrr6sR       r cluster_namecCtjtd|S)Nzcluster-{}.lockospathjoinrrr(rrr get_lock_pathpr/cCr))Nzcluster-{}.stater*r.rrrget_state_pathtr0r1cCs|S)Nr)rrrrbootstrap_localxsr2)rr+typingrrray._common.utilsr"ray.autoscaler._private.cli_loggerrrrstrrrrr/r1r2rrrrs  """:&