o $i89 @sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZddlZddlmZddlZddlmZmZddlmZmZmZmZmZddlmZddlmZm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)e*e+Z,ej-d kZ.d e/fddZ0ddZ1ddZ2e,fdeej3fddZ4ddZ5e,fdeej3d ee/fddZ6 d*de e e fde/dee e/d e e e ffddZ7de d e/fd d!Z8d"e d ee/fd#d$Z9e,fd"d%deej3d e e/e ffd&d'Z:Gd(d)d)e"Z;dS)+N)Path)AnyDictListOptional)FileLock)get_or_create_event_looptry_to_create_directory)create_conda_env_if_neededdelete_conda_envget_conda_activate_commandsget_conda_envsget_conda_info_json)RuntimeEnvContext)Protocol parse_uri)RuntimeEnvPlugin)parse_and_validate_conda)get_directory_size_bytesget_master_wheel_urlget_release_wheel_urlget_wheel_filenamentreturncCstjtjtjddS)Nr)ospathsplitray__file__rr[/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_private/runtime_env/conda.py_resolve_current_ray_path+sr!cCs"t}tj|d}t|dS)zFind the Ray setup_spec from the currently running Ray. This function works even when Ray is built from source with pip install -e. zsetup.py setup_spec)r!rrjoinrunpyrun_path)ray_source_python_path setup_py_pathrrr _get_ray_setup_spec5sr(cCs tjtjd}tt|S)z"`, or None if no GC required.condaNzconda://)rYzAconda field received by RuntimeEnvAgent must be str or dict, not rE)rRr_rHr`rm TypeErrortype__name__)rnrourirrr get_uris   rt RuntimeEnvcCsZt|}|dus Jt|d}|r|dg}n |dr"t}ng}t|t|}|S)zEReturns the conda spec with the Ray and `python` dependency inserted.NrWz ray[default]_inject_current_ray)reloads conda_configrX get_extensionr/rcrK)rnr0rYray_pipextra_pip_dependenciesrrr !_get_conda_dict_with_ray_inserteds    r|c @seZdZdZdefddZdedefddZd d deefd d Ze fd ede e j de fddZe fd e ed d dede j de f ddZe fdeed d dede e j fddZdS) CondaPluginro resources_dircCs8tj|d|_t|jtj|jd|_t|_dS)Nroz%ray-conda-installs-and-deletions.lock)rrr#_resources_dirr !_installs_and_deletions_file_lockr-_validated_named_conda_env)selfr~rrr __init__s   zCondaPlugin.__init__rlrcCstj|j|S)aGenerate a path from the hash of a conda or pip spec. The output path also functions as the name of the conda environment when using the `--prefix` option to `conda create` and `conda remove`. Example output: /tmp/ray/session_2021-11-03_16-33-59_356303_41018/runtime_resources /conda/ray-9a7972c3a75f55e976e620484f58410c920db091 )rrr#r)rrlrrr _get_path_from_hashs zCondaPlugin._get_path_from_hashrnrucCs|}|r |gSgS)zFReturn the conda URI from the RuntimeEnv if it exists, else return [].) conda_uri)rrnrrrr get_urisszCondaPlugin.get_urisrsr0cCs|d|t|\}}|tjkrtd|d|||}t|}t|jt ||d}Wdn1sSs&z6CondaPlugin.create.._create..zThe given conda environment 'z' from the runtime env z doesn't exist from the output of `conda info --json`. You can only specify an env that already exists. Please make sure to create an env  z"Setting up conda for runtime_env: rWz"Setting up conda environment with zenvironment.ymlr2rrv)r>r0z'Finished creating conda environment at )rrRr_rHrrr anyradddebug serializerrr|rrrrrr#rr<yamldumpr removeryrDr) conda_infoenvsrrlconda_env_namerYconda_yaml_filefiler0rnrrsrr _createCsR           z#CondaPlugin.create.._create) has_condarrun_in_executor)rrsrnrr0rlooprrr create9s 6zCondaPlugin.createuriscCsT|sdS|r|}n t|\}}||}d|_|jt|7_dS)Nr1)rrrrr py_executablecommand_prefixr )rrrnrr0rrrlrrr modify_context|s  zCondaPlugin.modify_contextN)rr __module__ __qualname__namerHrrrrdefault_loggerrloggingLoggerintrrrrrrrr r}sF   Hr})N)sr           % )    2