o -wi["+@sddlZddlZddlmZmZmZmZmZddlZddl m m Z ddl m Z ddlmZddlmZddlmZddlmZmZmZmZdd lmZ d)d e d ed eeefd eededeedefddZ                    d*deedeedeeeefd eeeeeeeffdeedeed eedeedeeedeedeedeedeedeeeefdeed ee d!eed"eed#eedeedd$f*d%d&Z!       d+d e deedeeeefd eedeedeed eedeedeeedeedeedeedeedeeeefdeed ee d!eed"eed#eedeedd$f*d'd(Z"dS),N)AnyDictListOptionalUnion)Api) CommError)build_image_from_project) LaunchError)LAUNCH_DEFAULT_PROJECT LOG_PREFIXconstruct_launch_specvalidate_launch_spec_source) LaunchProjectapi queue_name launch_spectemplate_variables project_queuepriorityreturncCs||||||SN)push_to_run_queue)rrrrrrrY/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/launch/_launch_add.py push_to_queues rFurijobconfigprojectentityresource entry_pointnameversion docker_image resource_argsrun_idbuild repositorysweep_idauthorzpublic.QueuedRuncCs6t}t|||||||||| | | | | ||||||dS)a Enqueue a W&B launch experiment. With either a source uri, job or docker_image. Arguments: uri: URI of experiment to run. A wandb run uri or a Git repository URI. job: string reference to a wandb.Job eg: wandb/test/my-job:latest config: A dictionary containing the configuration for the run. May also contain resource specific arguments under the key "resource_args" template_variables: A dictionary containing values of template variables for a run queue. Expected format of `{"VAR_NAME": VAR_VALUE}` project: Target project to send launched run to entity: Target entity to send launched run to queue: the name of the queue to enqueue the run to priority: the priority level of the job, where 1 is the highest priority resource: Execution backend for the run: W&B provides built-in support for "local-container" backend 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. name: Name run under which to launch the run. version: For Git-based projects, either a commit hash or a branch name. docker_image: The name of the docker image to use 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``. run_id: optional string indicating the id of the launched run build: optional flag defaulting to false, requires queue to be set if build, an image is created, creates a job artifact, pushes a reference to that job artifact to queue repository: optional string to control the name of the remote repository, used when pushing images to a registry project_queue: optional string to control the name of the project for the queue. Primarily used for back compatibility with project scoped queues Example: ```python from wandb.sdk.launch import launch_add project_uri = "https://github.com/wandb/examples" params = {"alpha": 0.5, "l1_ratio": 0.01} # Run W&B project and create a reproducible docker environment # on a local host api = wandb.apis.internal.Api() launch_add(uri=project_uri, parameters=params) ``` Returns: an instance of`wandb.api.public.QueuedRun` which gives information about the queued run, or if `wait_until_started` or `wait_until_finished` are called, gives access to the underlying Run information. Raises: `wandb.exceptions.LaunchError` if unsuccessful )r(r)r*r+r,r)r _launch_add)rrrrr r!rr"r#r$r%r&rr'r(r)r*r+r,rrrrr launch_add"s.Jr.c Csrtd||| ||| ||| | |||||}|ro|dkrtd|ddur-tdd|d<t||}tt |||p;i}tjpLtj |d|ddd}| ||j }| j}|dd |dd |}||d<d|d <|durud }| dur{t} |d }t|tr|d |dur|}ni||}t|t||||| |}|dusd |vrtd|d}|r|dr|d|d<|dr|d|d<| tkrttd|dnttd| d |dttdt|dt}|dur'z |j|ddWnttfy&}z td|d|d}~ww||d|d||d | |}|S)Nz local-processz;Cannot build a docker image for the resource: local-processrz5Build doesn't support setting a job. Overwriting job.r r! launch_job)r r!job_type/rdefaultrrunQueueItemIdzError adding run to queuerunSpecr'r"zAdded run to queue .z Launch spec:  )typezUnable to fetch job with name z: )r r getwandbtermwarnr from_specasynciorunr init_log_job_artifact_with_image override_argswaitr$r isinstancedictpoprrtermlogr pprintpformatpublicr _artifact ValueErrorr queued_run) rrrrr r!rr"r#r$r%r&rr'r(r)r*r+r,rrlaunch_projectdocker_image_urir= job_artifactjob_namespec_template_varsres updated_spec public_apierKrrrr-s            r-r)NNNNNNNNNNNNNNNFNNNN)NNFNNNN)#r<rFtypingrrrrrr9wandb.apis.publicapisrHwandb.apis.internalr wandb.errorsrwandb.sdk.launch.builder.buildr wandb.sdk.launch.errorsr wandb.sdk.launch.utilsr r r r _project_specrstrrCintrfloatboolr.r-rrrrs              r