| 1234567891011121314151617181920212223242526272829 |
- from typing_extensions import Self
- from wandb.docker import is_docker_installed
- from wandb.sdk.launch.utils import docker_image_exists
- from .abstract import AbstractRegistry
- class AnonynmousRegistry(AbstractRegistry):
- def __init__(self, uri: str) -> None:
- """Initialize the registry."""
- self.uri = uri
- async def get_username_password(self) -> tuple[str, str]:
- """Get the username and password for the registry."""
- raise NotImplementedError("Anonymous registry does not require authentication")
- async def get_repo_uri(self) -> str:
- return self.uri
- async def check_image_exists(self, image_uri: str) -> bool:
- """Check if an image exists in the registry."""
- if not is_docker_installed():
- return False
- return docker_image_exists(image_uri)
- @classmethod
- def from_config(cls, config: dict) -> Self:
- return cls(uri=config["uri"])
|
