o xi!@sddlmZddlZddlZddlmZddlZddlmm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZmZmZd d lmZ d5d6ddZ                    d7d8d1d2Z       d9d:d3d4ZdS);) annotationsN)Any)Api) CommError)build_image_from_project) LaunchError)LAUNCH_DEFAULT_PROJECT LOG_PREFIXconstruct_launch_specvalidate_launch_spec_source) LaunchProjectapir queue_namestr launch_specdict[str, Any]template_variables dict | None project_queuepriority int | NonereturnrcCs||||||SN)push_to_run_queue)rrrrrrrP/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/_launch_add.py push_to_queues rFuri str | Nonejobconfigdict[str, Any] | None#dict[str, float | int | str] | Noneprojectentityresource entry_pointlist[str] | Nonenameversion docker_image resource_argsrun_idbuild bool | None repositorysweep_idauthorpublic.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.r0r1r2r)r _launch_add)rr r!rr$r%rr&r'r)r*r+rr,r-r.r0r1r2rrrrr launch_add$s.Jr5c 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-processr z5Build 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 rgetwandbtermwarnr from_specasynciorunrinit_log_job_artifact_with_image override_argswaitr)r isinstancedictpopr rtermlogr pprintpformatpublicr _artifact ValueErrorr queued_run) rr r!rr$r%rr&r'r)r*r+rr,r-r.r0r1r2rrlaunch_projectdocker_image_urirD job_artifactjob_namespec_template_varsres updated_spec public_apierRrrrr4s            r4r)rrrrrrrrrrrrrr)NNNNNNNNNNNNNNNFNNNN)*rrr rr!r"rr#r$rr%rrrr&rr'r(r)rr*rr+rrrr,r"r-rr.r/r0rr1rr2rrrrr3)NNFNNNN)*rrr rr!r"rrr$rr%rrrr&rr'r(r)rr*rr+rrrr,r"r-rr.r/r0rr1rr2rrrrr3) __future__rrCrMtypingrr@wandb.apis.publicapisrOwandb.apis.internalr wandb.errorsrwandb.sdk.launch.builder.buildrwandb.sdk.launch.errorsrwandb.sdk.launch.utilsrr r r _project_specr rr5r4rrrrsV         r