nemo_gym.orchestration.executors.connection

Module Contents

Classes

Name	Description
`Connection`	-
`LocalConnection`	-
`SSHConnection`	Single SSH master connection; copy and all commands reuse the same socket.

Functions

Name	Description
`_checked`	-
`get_connection`	-

API

class nemo_gym.orchestration.executors.connection.Connection()

Abstract

nemo_gym.orchestration.executors.connection.Connection.__enter__() -> nemo_gym.orchestration.executors.connection.Connection

nemo_gym.orchestration.executors.connection.Connection.__exit__(
    _ = ()
) -> None

nemo_gym.orchestration.executors.connection.Connection.close() -> None

nemo_gym.orchestration.executors.connection.Connection.copy(
    local: pathlib.Path,
    remote: pathlib.Path
) -> None

abstract

nemo_gym.orchestration.executors.connection.Connection.run(
    commands: list[str]
) -> str

abstract

nemo_gym.orchestration.executors.connection.Connection.write_text(
    remote: pathlib.Path,
    content: str,
    private: bool = False
) -> None

abstract

class nemo_gym.orchestration.executors.connection.LocalConnection()

Bases: Connection

nemo_gym.orchestration.executors.connection.LocalConnection.copy(
    local: pathlib.Path,
    remote: pathlib.Path
) -> None

nemo_gym.orchestration.executors.connection.LocalConnection.run(
    commands: list[str]
) -> str

nemo_gym.orchestration.executors.connection.LocalConnection.write_text(
    remote: pathlib.Path,
    content: str,
    private: bool = False
) -> None

class nemo_gym.orchestration.executors.connection.SSHConnection(
    hostname: str
)

Bases: Connection

Single SSH master connection; copy and all commands reuse the same socket.

_MASTER_TIMEOUT

= 10

_OPTS

_master

Popen | None = None

_socket

nemo_gym.orchestration.executors.connection.SSHConnection.__enter__() -> nemo_gym.orchestration.executors.connection.SSHConnection

nemo_gym.orchestration.executors.connection.SSHConnection._ssh_opts() -> list[str]

nemo_gym.orchestration.executors.connection.SSHConnection._wait_for_socket() -> None

nemo_gym.orchestration.executors.connection.SSHConnection.close() -> None

nemo_gym.orchestration.executors.connection.SSHConnection.copy(
    local: pathlib.Path,
    remote: pathlib.Path
) -> None

nemo_gym.orchestration.executors.connection.SSHConnection.run(
    commands: list[str]
) -> str

nemo_gym.orchestration.executors.connection.SSHConnection.write_text(
    remote: pathlib.Path,
    content: str,
    private: bool = False
) -> None

nemo_gym.orchestration.executors.connection._checked(
    cmd: list[str],
    input: str | None = None,
    context: str = ''
) -> str

nemo_gym.orchestration.executors.connection.get_connection(
    hostname: str | None
) -> nemo_gym.orchestration.executors.connection.Connection