File size: 943 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from typing import Tuple
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) -> "AbstractRegistry":
return cls(uri=config["uri"])
|