o wii@s6dZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZddlZddlmZddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZm Z m!Z!m"Z"e rxddl#m$Z$m%Z%ddl&m'Z'GdddZ(GdddZ)e dZ*e dZ+e dZ,GdddZ-dS)zW&B Public API for management Launch Jobs and Launch Queues. This module provides classes for managing W&B jobs, queued runs, and run queues. ) annotationsN)Mapping) TYPE_CHECKINGAnyCallableLiteral)gql)util)public)normalize_exceptions) CommError) InvalidTypeType TypeRegistry) LaunchError)LAUNCH_DEFAULT_PROJECT_fetch_git_repo apply_patch"convert_jupyter_notebook_to_script)ApiRetryingClient) LaunchProjectc@seZdZUded<ded<ded<ded<ded<ded <d ed <d ed <d.d/ddZed0ddZd1ddZddZd2d d!Z d2d"d#Z d2d$d%Z d2d&d'Z d3d)d*Z + d4d,d-Zd S)5Jobstr_namer _input_types _output_types_entity_project list[str] _entrypointbool _notebook_job_partialNapirpath str | NonereturnNonecCsz |j|dd|_Wntytd|dw|r'||_|jj|dn|j|_||_||_|j|_t t j |jd}t ||_Wdn1sRwY|jdi}|dd |_|d |_|d |_|d |_|d |_|d|_|jdd |_t j |jd|_t|jd|_t|jd|_|jddkr||j|jddkr||j|jddkr||j dSdS)Njob)typez Job artifact z not found)rootzwandb-job.jsonsourcenotebookF entrypoint dockerfile build_context base_imageargsr#zrequirements.frozen.txt input_types output_types source_typeartifactrepoimage)! _artifact _job_artifactr _fpathdownloadr_apidefault_entityropenosr%joinjsonload _job_infogetr"r _dockerfile_build_context _base_image_argsr#_requirements_filertype_from_dictrr_set_configure_launch_project"_configure_launch_project_artifact_configure_launch_project_repo#_configure_launch_project_container)selfr$namer%f source_inforTJ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/apis/public/jobs.py__init__/sJ           z Job.__init__cC|jS)zThe name of the job.rrPrTrTrUrQVzJob.namefuncCallable[[LaunchProject], None]cCs ||_dSN)configure_launch_project)rPr[rTrTrUrL[s z!Job._set_configure_launch_projectcCs|ddlm}t|\}}}|rtj||jj}n|jj |dd}|dur*t d|j |j krZdS)JRunQueueaClass that represents a run queue in W&B. Args: client: W&B API client instance. name: Name of the run queue entity: The entity (user or team) that owns this queue prioritization_mode: Queue priority mode Can be "DISABLED" or "V0". Defaults to `None`. _access: Access level for the queue Can be "project" or "user". Defaults to `None`. _default_resource_config_id: ID of default resource config _default_resource_config: Default resource configuration NrrrQrrprioritization_mode!RunQueuePrioritizationMode | None_accessRunQueueAccessType | None_default_resource_config_idr_default_resource_configdict[str, Any] | Noner'r(cCsF||_||_||_||_||_||_||_d|_d|_d|_ d|_ dSr]) rrgr_prioritization_moderrr_template_variables_type_items_id)rPrrQrrrrrrTrTrUrVs  zRunQueue.__init__cCrWrrXrYrTrTrUrQrZz RunQueue.namecCrW)zThe entity that owns the queue.rrYrTrTrUr rZzRunQueue.entityRunQueuePrioritizationModecC|jdur ||jS)zYThe prioritization mode of the queue. Can be set to "DISABLED" or "V0". N)r _get_metadatarYrTrTrUrs zRunQueue.prioritization_modeRunQueueAccessTypecCr)zThe access level of the queue.N)rrrYrTrTrUaccess zRunQueue.accessdict[str, str]cCr)z&External resource links for the queue.N)_external_linksrrYrTrTrUexternal_links#rzRunQueue.external_linksRunQueueResourceTypecC*|jdur|jdur|||jS)z The resource type for execution.N)rrr_get_default_resource_configrYrTrTrUr**  z RunQueue.typercCr)z(The default configuration for resources.N)rrrrrYrTrTrUdefault_resource_config3rz RunQueue.default_resource_configcCr)z!Variables for resource templates.N)rrrrrYrTrTrUr<rzRunQueue.template_variablescCr)zThe id of the queue.N)rrrYrTrTrUrErz RunQueue.idlist[QueuedRun]cCr)zeUp to the first 100 queued runs. Modifying this list will not modify the queue or any enqueued items!N)r _get_itemsrYrTrTrUrLs zRunQueue.itemscCs^td}d|ji}|j||}|ddr'd|_d|_d|_d|_d|_dSt d|j )z,Delete the run queue from the wandb backend.z mutation DeleteRunQueue($id: ID!) { deleteRunQueues(input: {queueIDs: [$id]}) { success clientMutationId } } rdeleteRunQueuessuccessNzFailed to delete run queue ) rrrgrrrrrrr rQrPrrrrTrTrUrTs   zRunQueue.deletecCr)Nz )rrrYrTrTrUrlrzRunQueue.__repr__cCstd}t|j|jd}|j||}|ddd|_|ddd|_|ddd|_|ddd|_ |jdur?i|_ |ddd |_ dS) Na query GetRunQueueMetadata($projectName: String!, $entityName: String!, $runQueue: String!) { project(name: $projectName, entityName: $entityName) { runQueue(name: $runQueue) { id access defaultResourceConfigID prioritizationMode externalLinks } } } rrrrrdefaultResourceConfigID externalLinksprioritizationMode) rrrrrgrrrrrrrrrTrTrUros"  zRunQueue._get_metadatacCs^td}|j|jd}|j||}|ddd|_|ddd|_|ddd|_dS)Na query GetDefaultResourceConfig($entityName: String!, $id: ID!) { entity(name: $entityName) { defaultResourceConfig(id: $id) { config resource templateVariables { name schema } } } } )rrrdefaultResourceConfigrrtemplateVariables)rrrrgrrrrrrTrTrUrs  z%RunQueue._get_default_resource_configc Csptd}t|j|jd}|j||}g|_|ddddD]}|jt|j|jt|j|ddq dS) Na query GetRunQueueItems($projectName: String!, $entityName: String!, $runQueue: String!) { project(name: $projectName, entityName: $entityName) { runQueue(name: $runQueue) { runQueueItems(first: 100) { edges { node { id } } } } } } rrrrrrr) rrrrrgrrappendrrrTrTrUrs( zRunQueue._get_itemsrr&r dict | Nonerc Cs&ddlm}|}|||||||S)aCreate a RunQueue. Args: name: The name of the run queue to create. resource: The resource type for execution. entity: The entity (user or team) that will own the queue. Defaults to the default entity of the API client. prioritization_mode: The prioritization mode for the queue. Can be "DISABLED" or "V0". Defaults to None. config: Optional dictionary for the default resource configuration. Defaults to None. template_variables: Optional dictionary for template variables used in the resource configuration. r)r)wandb.apis.publicrcreate_run_queue) clsrQrrrrrr public_apirTrTrUcreates  zRunQueue.create)NNNN)rrrQrrrrrrrrrrrr'r(r)r'r)r'r)r'r)r'rr)r'r)r'r()rQrrrrr&rrrrrrr'r)rrrrrVrrQrrrrr*rrrrr rrrrr classmethodrrTrTrTrUrsT                $r).r __future__rrBr@r|rcollections.abcrtypingrrrr wandb_gqlrrdr wandb.apisr wandb.apis.normalizer wandb.errorsr wandb.sdk.data_types._dtypesr rrwandb.sdk.launch.errorsrwandb.sdk.launch.utilsrrrrrrrwandb.sdk.launch._project_specrrrrrrrrTrTrTrUs<         Vc