nerfbaselines.backends

class nerfbaselines.backends.ApptainerBackend(spec: ApptainerBackendSpec)

Bases: RemoteProcessRPCBackend

install()

name = 'apptainer'

shell(args=None)

class nerfbaselines.backends.ApptainerBackendSpec

Bases: TypedDict

conda_spec: CondaBackendSpec | None

default_cuda_archs: str

environment_name: Required[str]

home_path: str

image: str | None

python_path: str

class nerfbaselines.backends.Backend

Bases: object

install()

instance_call(instance: int, method: str, *args, **kwargs)

instance_del(instance: int)

name = 'unknown'

shell(args: Tuple[str, ...] | None = None)

static_call(function: str, *args, **kwargs)

class nerfbaselines.backends.CondaBackend(spec: CondaBackendSpec)

Bases: RemoteProcessRPCBackend

install()

name = 'conda'

shell(args=None)

class nerfbaselines.backends.CondaBackendSpec

Bases: TypedDict

environment_name: Required[str]

install_script: str | None

python_version: str | None

class nerfbaselines.backends.DockerBackend(spec: DockerBackendSpec)

Bases: RemoteProcessRPCBackend

install()

name = 'docker'

shell(args=None)

class nerfbaselines.backends.DockerBackendSpec

Bases: TypedDict

build_script: str

conda_spec: CondaBackendSpec

default_cuda_archs: str

environment_name: Required[str]

home_path: str

image: str | None

python_path: str

replace_user: bool

class nerfbaselines.backends.SimpleBackend

Bases: Backend

instance_call(instance: int, method: str, *args, **kwargs)

instance_del(instance: int)

name = 'python'

static_call(function: str, *args, **kwargs)

nerfbaselines.backends.forward_port(ps: int, pd: int)

nerfbaselines.backends.get_backend(method_spec: MethodSpec, backend: str | None) → Backend

nerfbaselines.backends.get_forwarded_ports()

nerfbaselines.backends.get_implemented_backends(method_spec: MethodSpec) → Sequence[Literal['conda', 'docker', 'apptainer', 'python']]

nerfbaselines.backends.get_mounts()

nerfbaselines.backends.mount(ps: str | Path, pd: str | Path)

nerfbaselines.backends.run_on_host()

nerfbaselines.backends.protocol_shm_pickle

exception nerfbaselines.backends.protocol_shm_pickle.ConnectionClosed

Bases: ConnectionError

class nerfbaselines.backends.protocol_shm_pickle.SharedMemoryProtocol(*, shared_memory_name=None, shared_memory_size=134217728)

Bases: object

close()

connect_worker()

get_worker_configuration()

property protocol_name

receive(interrupt=False)

send(message, interrupt=False)

start_host()

wait_for_worker(timeout=None)

nerfbaselines.backends.protocol_tcp_pickle

class nerfbaselines.backends.protocol_tcp_pickle.TCPPickleProtocol(*, hostname=None, port=None, authkey=None, max_message_size=134217728)

Bases: object

close()

connect_worker()

get_worker_configuration()

property protocol_name

receive(interrupt=False)

send(message, interrupt=False)

start_host()

wait_for_worker(timeout=None)