nemo_gym.sandbox.providers.registry

View as Markdown

Provider registration utilities.

Module Contents

Functions

NameDescription
_load_apptainer_provider-
_load_opensandbox_provider-
create_providerInstantiate a provider from a single-key provider config.
get_provider_classReturn a registered provider class.
list_providersList registered provider names.
register_providerRegister a sandbox provider class.

Data

ProviderClass

ProviderLoader

_BUILTIN_PROVIDER_LOADERS

_PROVIDER_REGISTRY

API

nemo_gym.sandbox.providers.registry._load_apptainer_provider() -> nemo_gym.sandbox.providers.registry.ProviderClass
nemo_gym.sandbox.providers.registry._load_opensandbox_provider() -> nemo_gym.sandbox.providers.registry.ProviderClass
nemo_gym.sandbox.providers.registry.create_provider(
config: collections.abc.Mapping[str, typing.Any]
) -> nemo_gym.sandbox.providers.base.SandboxProvider

Instantiate a provider from a single-key provider config.

nemo_gym.sandbox.providers.registry.get_provider_class(
name: str
) -> nemo_gym.sandbox.providers.registry.ProviderClass

Return a registered provider class.

nemo_gym.sandbox.providers.registry.list_providers() -> list[str]

List registered provider names.

nemo_gym.sandbox.providers.registry.register_provider(
name: str,
provider_class: nemo_gym.sandbox.providers.registry.ProviderClass,
override: bool = False
) -> None

Register a sandbox provider class.

nemo_gym.sandbox.providers.registry.ProviderClass: TypeAlias = type[SandboxProvider]
nemo_gym.sandbox.providers.registry.ProviderLoader: TypeAlias = Callable[[], ProviderClass]
nemo_gym.sandbox.providers.registry._BUILTIN_PROVIDER_LOADERS: dict[str, ProviderLoader] = {}
nemo_gym.sandbox.providers.registry._PROVIDER_REGISTRY: dict[str, ProviderClass] = {}