Source code for wsipipe.load.datasets.registry

from wsipipe.load.datasets.loader import Loader
from wsipipe.load.datasets.camelyon16 import Camelyon16Loader
from wsipipe.load.datasets.stripai import StripaiLoader


loaders = {}


def get_loader(name: str) -> Loader:
    """A convenience function to call loader based on its name

    Args:
        name (str): Name of the loader.
    Returns:
        loader (Loader): the loader class
    """
    # print(f"calling get_loader({name})")
    if name in loaders:
        return loaders[name]
    else:
        constructor = eval(name)
        loader = constructor()
        loaders[name] = loader
        return loader