Skip to content

pixano_inference.providers.utils

Provider for the SAM2 model.

get_provider(provider)

Return the provider from the registry.

Source code in pixano_inference/providers/utils.py 
13
14
15
16
17
def get_provider(provider: str) -> type[BaseProvider]:
    """Return the provider from the registry."""
    if (actual_provider := PROVIDERS_REGISTRY.get(provider)) is not None:
        return actual_provider
    raise ValueError(f"Provider {provider} not found.")

get_provider_name(provider)

Return the name of the provider.

Source code in pixano_inference/providers/utils.py 
30
31
32
33
34
35
def get_provider_name(provider: BaseProvider) -> str:
    """Return the name of the provider."""
    for name, provider_cls in PROVIDERS_REGISTRY.items():
        if isinstance(provider, provider_cls):
            return name
    raise ValueError(f"Provider {provider} not found.")

get_providers()

Return the list of providers in the registry.

Source code in pixano_inference/providers/utils.py 
20
21
22
def get_providers() -> list[str]:
    """Return the list of providers in the registry."""
    return list(PROVIDERS_REGISTRY.keys())

instantiate_provider(provider)

Instantiate a provider.

Source code in pixano_inference/providers/utils.py 
38
39
40
def instantiate_provider(provider: str) -> BaseProvider:
    """Instantiate a provider."""
    return get_provider(provider)()

is_provider(provider)

Return True if the provider is in the registry.

Source code in pixano_inference/providers/utils.py 
25
26
27
def is_provider(provider: str) -> bool:
    """Return True if the provider is in the registry."""
    return provider in PROVIDERS_REGISTRY