o xi"@sdZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd lmZdd lmZdd lmZhd Zd%ddZd&ddZd'ddZd(d#d$ZdS))zUtilities for the agent.) annotations)AnyN)Api)is_docker_installed) LaunchError)AbstractBuilder)AbstractEnvironment)AbstractRegistry)AbstractRunner>vertex sagemaker kubernetes local-processlocal-containerconfigdict[str, Any] | Nonereturnr cCs|s ddlm}|S|d}|std|dkr%ddlm}||S|dkr4ddlm}||S|dkrCdd lm}||S|d krRdd l m }||Std |) aCreate an environment from a config. This helper function is used to create an environment from a config. The config should have a "type" key that specifies the type of environment to create. The remaining keys are passed to the environment's from_config method. If the config is None or empty, a LocalEnvironment is returned. Arguments: config (Dict[str, Any]): The config. Returns: Environment: The environment constructed. r)LocalEnvironmenttypezICould not create environment from config. Environment type not specified!localawsAwsEnvironmentgcpGcpEnvironmentazure)AzureEnvironmentz8Could not create environment from config. Invalid type: ) environment.local_environmentrgetr from_configenvironment.aws_environmentrenvironment.gcp_environmentrenvironment.azure_environmentr)rrenv_typerrrr&K/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/loader.pyenvironment_from_configs.          r( environmentr cCs|s ddlm}|Std|d}|dus|dkr&ddlm}|S|dkr5ddlm}||S|d krDdd lm }||S|d krSdd l m }||St d |)anCreate a registry from a config. This helper function is used to create a registry from a config. The config should have a "type" key that specifies the type of registry to create. The remaining keys are passed to the registry's from_config method. If the config is None or empty, a LocalRegistry is returned. Arguments: config (Dict[str, Any]): The registry config. environment (Environment): The environment of the registry. Returns: The registry if config is not None, otherwise None. Raises: LaunchError: If the registry is not configured correctly. r) LocalRegistryaThe `registry` block of the launch agent config is being deprecated. Please specify an image repository URI under the `builder.destination` key of your launch agent config. See https://docs.wandb.ai/guides/launch/setup-agent-advanced#agent-configuration for more information.rNrecr)ElasticContainerRegistrygcr)GoogleArtifactRegistryacr)AzureContainerRegistryz>Could not create registry from config. Invalid registry type: ) registry.local_registryr*wandbtermwarnr #registry.elastic_container_registryr,r!!registry.google_artifact_registryr.!registry.azure_container_registryr0r)rr)r* registry_typer,r.r0r&r&r'registry_from_configFs,         r8registryrcCs|strddlm}|i||Sddlm}|i||S|d}|dur,td|dkr=ddlm}||||S|dkrNdd lm }||||S|d kr_ddlm}||||Std |) aCreate a builder from a config. This helper function is used to create a builder from a config. The config should have a "type" key that specifies the type of builder to import and create. The remaining keys are passed to the builder's from_config method. If the config is None or empty, a default builder is returned. The default builder will be a DockerBuilder if we find a working docker cli on the system, otherwise it will be a NoOpBuilder. Arguments: config (Dict[str, Any]): The builder config. registry (Registry): The registry of the builder. Returns: The builder. Raises: LaunchError: If the builder is not configured correctly. r) DockerBuilder) NoOpBuilderrNz@Could not create builder from config. Builder type not specifieddockerkaniko) KanikoBuildernoopz)rr)r9r:r; builder_typer>r&r&r'builder_from_config}s2      rD runner_namestrapir runner_configdict[str, Any]r c CsD|r|dvrddlm}|||||S|dkr"ddlm}|||S|dkrVddlm}t||sIz|}WntyH}ztd|d }~wwdd l m } | ||||S|d vrdd l m } t|| s}z| }Wnty|}ztd |d }~wwddl m} | ||||S|dkrddlm} | ||||Std|)a`Create a runner from a config. This helper function is used to create a runner from a config. The config should have a "type" key that specifies the type of runner to import and create. The remaining keys are passed to the runner's from_config method. If the config is None or empty, a LocalContainerRunner is returned. Arguments: runner_name (str): The name of the backend. api (Api): The API. runner_config (Dict[str, Any]): The backend config. Returns: The runner. Raises: LaunchError: If the runner is not configured correctly. )rrr)LocalContainerRunnerr)LocalProcessRunnerr rzUCould not create Sagemaker runner. Environment must be an instance of AwsEnvironment.N)SageMakerRunner)r z gcp-vertexrzRCould not create Vertex runner. Environment must be an instance of GcpEnvironment.) VertexRunnerr)KubernetesRunnerz:Could not create runner from config. Invalid runner name: )runner.local_containerrJrunner.local_processrKr"r isinstance from_defaultrrunner.sagemaker_runnerrLr#rrunner.vertex_runnerrMrunner.kubernetes_runnerrN) rErGrHr)r9rJrKrerLrrMrNr&r&r'runner_from_configsT             rW)rrrr )rrr)r rr )rrr)r r9r rr) rErFrGrrHrIr)r r9r rr )__doc__ __future__rtypingrr2wandb.apis.internalr wandb.dockerrwandb.sdk.launch.errorsrbuilder.abstractrenvironment.abstractr registry.abstractr runner.abstractr WANDB_RUNNERSr(r8rDrWr&r&r&r's           , 7;