o xi@sdZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZdd lmZdd lmZmZmZerDdd lmZGd d d eZdddZdS)]Abstract plugin class defining the interface needed to build container images for W&B Launch.) annotations)ABCabstractmethod) TYPE_CHECKINGAny)AbstractEnvironment)AbstractRegistry) EntryPoint LaunchProject)AnonynmousRegistry)"AZURE_CONTAINER_REGISTRY_URI_REGEX$ELASTIC_CONTAINER_REGISTRY_URI_REGEXGCP_ARTIFACT_REGISTRY_URI_REGEX)JobAndRunStatusTrackerc@sveZdZUdZded<ded<ded<ded <e d d!ddZeed"ddZe d#d$ddZ ed%ddZ dS)&AbstractBuilderrstr builder_typer environmentr registryzdict[str, Any]builder_configTverifyboolreturnNonecCt)a1Initialize a builder. Arguments: builder_config: The builder config. registry: The registry to use. verify: Whether to verify the functionality of the builder. Raises: LaunchError: If the builder cannot be initialized or verified. NotImplementedError)selfrrrr U/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/builder/abstract.py__init__szAbstractBuilder.__init__configdictcCr)aCreate a builder from a config dictionary. Arguments: config: The config dictionary. environment: The environment to use. registry: The registry to use. verify: Whether to verify the functionality of the builder. login: Whether to login to the registry immediately. Returns: The builder. r)clsr#rrr r r! from_config2szAbstractBuilder.from_configNlaunch_projectr entrypointr job_trackerJobAndRunStatusTracker | Nonect)zBuild the image for the given project. Arguments: launch_project: The project to build. build_ctx_path: The path to the build context. Returns: The image name. r)rr'r(r)r r r! build_imageHszAbstractBuilder.build_imagecr+)zVerify that the builder can be used to build images. Raises: LaunchError: If the builder cannot be used to build images. r)rr r r!rZszAbstractBuilder.verify)T)rrrr rrrr)r#r$rrrr rr)N)r'r r(r r)r*rr)rr) __name__ __module__ __qualname____doc____annotations__rr" classmethodr&r,rr r r r!rs" rurirrr cCs|dr |tdd}t|durddlm}||dSt|dur1ddlm}||dSt |durCddl m }||dSt |dS)aJCreate a registry helper object from a uri. This function parses the URI and determines which supported registry it belongs to. It then creates a registry helper object for that registry. The supported remote registry types are: - Azure Container Registry - Google Container Registry - AWS Elastic Container Registry The format of the URI is as follows: - Azure Container Registry: .azurecr.io// - Google Container Registry: -docker.pkg.dev/// - AWS Elastic Container Registry: .dkr.ecr..amazonaws.com// Our classification of the registry is based on the domain name. For example, if the uri contains `.azurecr.io`, we classify it as an Azure Container Registry. If the uri contains `.dkr.ecr`, we classify it as an AWS Elastic Container Registry. If the uri contains `-docker.pkg.dev`, we classify it as a Google Artifact Registry. This function will attempt to load the appropriate cloud helpers for the `https://` prefix is optional for all of the above. Arguments: uri: The uri to create a registry from. Returns: The registry. Raises: LaunchError: If the registry helper cannot be loaded for the given URI. zhttps://Nr)AzureContainerRegistry)r3)GoogleArtifactRegistry)ElasticContainerRegistry) startswithlenrmatch2wandb.sdk.launch.registry.azure_container_registryr4r2wandb.sdk.launch.registry.google_artifact_registryr5r4wandb.sdk.launch.registry.elastic_container_registryr6r )r3r4r5r6r r r!registry_from_urids "       r=N)r3rrr )r0 __future__rabcrrtypingrr%wandb.sdk.launch.environment.abstractr"wandb.sdk.launch.registry.abstractr _project_specr r registry.anonr utilsrrr)wandb.sdk.launch.agent.job_status_trackerrrr=r r r r!s     M