nemo_gym.server_runtime.registry_server

The head server: the registry every Gym server and client asks for the run’s config and server addresses.

Module Contents

Classes

Name	Description
`HeadServer`	-
`ServerInstanceDisplayConfig`	-

Data

DEFAULT_HEAD_SERVER_PORT

API

class nemo_gym.server_runtime.registry_server.HeadServer()

Bases: BaseServer

_cached_yaml

Optional[str] = None

_ready

bool = PrivateAttr(default=False)

_server_instances

List[dict] = []

config

BaseServerConfig

nemo_gym.server_runtime.registry_server.HeadServer.get_server_instances() -> typing.List[dict]

nemo_gym.server_runtime.registry_server.HeadServer.global_config_dict_yaml() -> str

async

nemo_gym.server_runtime.registry_server.HeadServer.invalidate_global_config_dict_yaml_cache() -> None

Clear the serialized global config cache.

nemo_gym.server_runtime.registry_server.HeadServer.mark_ready() -> None

nemo_gym.server_runtime.registry_server.HeadServer.run_webserver() -> typing.Tuple[uvicorn.Server, threading.Thread, nemo_gym.server_runtime.registry_server.HeadServer]

classmethod

nemo_gym.server_runtime.registry_server.HeadServer.set_server_instances(
    instances: typing.List
) -> None

nemo_gym.server_runtime.registry_server.HeadServer.setup_webserver() -> fastapi.FastAPI

class nemo_gym.server_runtime.registry_server.ServerInstanceDisplayConfig()

Bases: BaseModel

config_path

Optional[str] = None

dir_path

Optional[Path] = None

entrypoint

Optional[str] = None

host

Optional[str] = None

name

Optional[str] = None

pid

Optional[int] = None

port

Optional[int] = None

process_name

Optional[str] = None

server_type

Optional[str] = None

start_time

Optional[float] = None

status

Optional[ServerStatus] = None

uptime_seconds

Optional[float] = None

url

Optional[str] = None

nemo_gym.server_runtime.registry_server.DEFAULT_HEAD_SERVER_PORT = 11000