o -wi@sdZddlmZmZddlmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZdd lmZmZmZerBdd lmZGd d d eZd ede fddZdS)]Abstract plugin class defining the interface needed to build container images for W&B Launch.)ABCabstractmethod) TYPE_CHECKINGAnyDictOptional)AbstractEnvironment)AbstractRegistry) EntryPoint LaunchProject)AnonynmousRegistry)"AZURE_CONTAINER_REGISTRY_URI_REGEX$ELASTIC_CONTAINER_REGISTRY_URI_REGEXGCP_ARTIFACT_REGISTRY_URI_REGEX)JobAndRunStatusTrackerc @seZdZUdZeed<eed<eed<eee fed<e ddedede dd fd d Z e e d edededdfd dZe ddedededdefddZe dddZd S)AbstractBuilderr builder_type environmentregistrybuilder_configTverifyreturnNcCt)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^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/launch/builder/abstract.py__init__szAbstractBuilder.__init__configcCr)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!rrrrr from_config0szAbstractBuilder.from_configlaunch_project entrypoint job_trackerrct)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&rrr build_imageFszAbstractBuilder.build_imagecr')zVerify that the builder can be used to build images. Raises: LaunchError: If the builder cannot be used to build images. r)rrrrrXszAbstractBuilder.verify)T)N)rN)__name__ __module__ __qualname____doc__str__annotations__r r rrrboolr classmethoddictr#r r rr(rrrrrrsR   rurircCs|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)r2)GoogleArtifactRegistry)ElasticContainerRegistry) startswithlenrmatch2wandb.sdk.launch.registry.azure_container_registryr3r2wandb.sdk.launch.registry.google_artifact_registryr4r4wandb.sdk.launch.registry.elastic_container_registryr5r)r2r3r4r5rrrregistry_from_uribs "       r<N)r,abcrrtypingrrrr%wandb.sdk.launch.environment.abstractr "wandb.sdk.launch.registry.abstractr _project_specr r registry.anonrutilsrrr)wandb.sdk.launch.agent.job_status_trackerrrr-r<rrrrs    M