o ni@sdZddlZddlZddlmZmZmZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZmZdd lmZmZdd lmZdd lmZdd lmZmZm Z ddl!m"Z"m#Z#ddl$m%Z%e&e'Z(Gddde Z)dS)z%Implementation of the docker builder.N)AnyDictOptional)JobAndRunStatusTracker)AbstractBuilderregistry_from_uri)AbstractEnvironment)AbstractRegistry) EntryPoint LaunchProject)LaunchDockerError LaunchError)AnonynmousRegistry) LocalRegistry) LOG_PREFIXevent_loop_thread_exec$warn_failed_packages_from_build_logs)_WANDB_DOCKERFILE_NAMEvalidate_docker_installation)BuildContextManagerc @seZdZdZdZdZdededee e ffddZ e dee e fdeded dfd d Z dd dZdddZ ddededeed e fddZd S) DockerBuilderzxBuilds a docker image for a project. Attributes: builder_config (Dict[str, Any]): The builder config. dockerz linux/amd64 environmentregistryconfigcCs||_||_||_dS)aInitialize a DockerBuilder. Arguments: environment (AbstractEnvironment): The environment to use. registry (AbstractRegistry): The registry to use. Raises: LaunchError: If docker is not installed N)rrr)selfrrrre/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/launch/builder/docker_builder.py__init__(s zDockerBuilder.__init__returncCs@|d}|r|durttd|dt|}||||S)aCreate a DockerBuilder from a config. Arguments: config (Dict[str, Any]): The config. registry (AbstractRegistry): The registry to use. verify (bool, optional): Whether to verify the functionality of the builder. login (bool, optional): Whether to login to the registry. Returns: DockerBuilder: The DockerBuilder. destinationNz.Overriding registry from registry config with z from builder config.)getwandbtermwarnrr)clsrrr image_urirrr from_config;s  zDockerBuilder.from_configNcstIdHdS)zVerify the builder.N)r)rrrrverifyZszDockerBuilder.verifycszt|jtrttddSt|jtr!ttddS|jIdH\}}tt j }||||jj IdHdS)zLogin to the registry.z'No registry configured, skipping login.z#Anonymous registry, skipping login.N) isinstancerr_loggerinforrget_username_passwordrrloginuri)rusernamepasswordr.rrrr.^s   zDockerBuilder.loginlaunch_project entrypoint job_trackerc s|IdH|IdHt|d}|d\}}tj|t}|js'dn|j IdH}|r9|d|} n|j d|} | sP|j | IdHrP| St d| dzttj| g|||jddIdH} t| | |j|Wntjy} z|r|d td | d} ~ wwzt|Wntytd |d } t | Ynw|r| d\} }ttd | ttj| |IdH}|durt d|j!dkrd|d|vrt d|| S)zBuild the image for the given project. Arguments: launch_project (LaunchProject): The project to build. entrypoint (EntryPoint): The entrypoint to use. N)r2r:zimage z0 does not already exist in repository, building.platform)tagsfile context_pathr6buildz(Error communicating with docker client: zTemporary docker context file z was not deleted.zPushing image z"Failed to push image to repository sagemakerzThe push refers to repository []z'Unable to push image to ECR, response: )"r)r.rcreate_build_contextospathjoinrr get_repo_uri image_namebuild_requiredcheck_image_existsr+r,rrr:rr#rapi DockerError set_err_stager remove Exceptionrsplitr$termlogpushrresource)rr2r3r4build_context_managerbuild_ctx_path image_tag dockerfile repositoryr'outpute_msgregtag push_resprrr build_imageisd         zDockerBuilder.build_image)r!N)N)__name__ __module__ __qualname____doc__ builder_typetarget_platformrr rstrrr classmethodr(r)r.r r rrrYrrrrrsB     r)*r]loggingr>typingrrrr$ wandb.dockerr)wandb.sdk.launch.agent.job_status_trackerr!wandb.sdk.launch.builder.abstractrr%wandb.sdk.launch.environment.abstractr"wandb.sdk.launch.registry.abstractr _project_specr r errorsr r registry.anonrregistry.local_registryrutilsrrrr:rrcontext_managerr getLoggerrZr+rrrrrs&