o xi-@s4ddlmZddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZmZmZmZee Z!d9ddZ"d:ddZ#d;d"d#Z$             dd?d7d8Z&dS)@) annotationsN)Any)Api)loader) LaunchProject) LaunchAgent)construct_agent_configs)LocalEnvironment)ExecutionError LaunchError) AbstractRun)LAUNCH_CONFIG_FILEPROJECT_SYNCHRONOUSconstruct_launch_specvalidate_launch_spec_sourcelogfilestrreturnNonec Cstd}|dkr tj}n&zt|d}Wnty2}ztd|d|WYd}~dSd}~wwtd|dkrs z(resolve_agent_config..)default_settings)yamlospath expanduserrisfiler safe_load YAMLErrorr updateitemsenvironr=intlist isinstancerappendrdefault_entity)r.r0r2r4r5rDdefaults config_path launch_configfr)msgkeyssettingsapir+r@r,resolve_agent_config:sr      r[rZrdict[str, Any]c Cszddlm}Wn tytdwz |jdi|WnM|jyk}z@|}|D]5}ddd|dgD}d|}|d }t |t sT|d |d 7}|d |d 7}t |q-tdd}~wwt ||} z t| WdStjyYdSw)Nr)r4zewandb launch-agent requires pydantic to be installed. Please install with `pip install wandb[launch]`.cSsg|]}t|qSr+)r)r>xr+r+r, sz(create_and_run_agent..loczAgent config error in field inputz (value: )z: rWzInvalid launch agent configr+)wandb.sdk.launch.agentr4ModuleNotFoundErrorr AgentConfigValidationErrorerrorsjoinr=rPdictrrrasynciorunloopCancelledError) rZr4 agent_configr)rgerrorr`rWvalueagentr+r+r,create_and_run_agents8      rrjobnameproject docker_image entry_pointlist[str] | Noneversionresource resource_argsdict[str, Any] | NonerU synchronous bool | Nonerun_id repositoryr csV| duri} |dur d}td|||||||||| | | | dd}t|t||}||}|jp7|j}i}| |t<| pAi}t |\}}}t |}|dur^t |t s^|IdHt ||}t |||}|js|js|suJ|||dIdH}t |||||}|r|sJ|||IdH}|sJ|Std|ddt j)z[Helper that delegates to the project-running method corresponding to the passed-in backend.Nzlocal-container)authorzUnavailable backend z, available backends: z, )rrr from_specfetch_and_validate_projectget_job_entry_pointrvjob_base_imagerr renvironment_from_configrPr verifyregistry_from_configbuilder_from_config build_imagerunner_from_configrkr rh WANDB_RUNNERS)rZrsrtrur.rvrwryrzr{rUr}rr launch_speclaunch_project entrypoint image_uri runner_configr4environment_config build_configregistry_config environmentr8r9backend submitted_runr+r+r,_launchsf      rTcCs.tt||||| ||||| | || | d}|S)aLaunch a W&B launch experiment. Arguments: job: string reference to a wandb.Job eg: wandb/test/my-job:latest api: An instance of a wandb Api from wandb.apis.internal. entry_point: Entry point to run within the project. Defaults to using the entry point used in the original run for wandb URIs, or main.py for git repository URIs. version: For Git-based projects, either a commit hash or a branch name. name: Name run under which to launch the run. resource: Execution backend for the run. resource_args: Resource related arguments for launching runs onto a remote backend. Will be stored on the constructed launch config under ``resource_args``. project: Target project to send launched run to entity: Target entity to send launched run to config: A dictionary containing the configuration for the run. May also contain resource specific arguments under the key "resource_args". synchronous: Whether to block while waiting for a run to complete. Defaults to True. Note that if ``synchronous`` is False and ``backend`` is "local-container", this method will return, but the current process will block when exiting until the local run completes. If the current process is interrupted, any asynchronous runs launched via this method will be terminated. If ``synchronous`` is True and the run fails, the current process will error out as well. run_id: ID for the run (To ultimately replace the :name: field) repository: string name of repository path for remote registry Example: ```python from wandb.sdk.launch import launch job = "wandb/jobs/Hello World:latest" params = {"epochs": 5} # Run W&B project and create a reproducible docker environment # on a local host api = wandb.apis.internal.Api() launch(api, job, parameters=params) ``` Returns: an instance of`wandb.launch.SubmittedRun` exposing information (e.g. run ID) about the launched run. Raises: `wandb.exceptions.ExecutionError` If a run launched in blocking mode is unsuccessful. )rsrtrur.rvrwryrzr{rUr}rZrr)rjrkr)rZrsrwryrtrzr{rur.rvr4r}rrsubmitted_run_objr+r+r,launchs&?r)rrrr) r.r/r0r1r2r3r4r/r5r1rr6)rZrr4r\rr) NNNNNNNNNNNNN)rZrrsr/rtr/rur/r.r/rvr/rwrxryr/rzr/r{r|rUr|r}r~rr/rr/rr ) NNNNNNNNNNTNN)rZrrsr/rwrxryr/rtr/rzr/r{r|rur/r.r/rvr/r4r|r}r~rr/rr/rr )' __future__rrjrrErtypingrrwandb.apis.internalrr _project_specrrqr agent.agentr environment.local_environmentr rgr r runner.abstractr utilsrrrrr__name___loggerr-r[rrrrr+r+r+r,s`             V! N