mlos_bench.environments.remote.os_env

OS-level remote Environment on Azure.

Classes

OSEnv

OS Level Environment for a host.

Module Contents

class mlos_bench.environments.remote.os_env.OSEnv(*, name: str, config: dict, global_config: dict | None = None, tunables: mlos_bench.tunables.tunable_groups.TunableGroups | None = None, service: mlos_bench.services.base_service.Service | None = None)[source]

Bases: mlos_bench.environments.base_environment.Environment

OS Level Environment for a host.

Create a new environment for remote execution.

Parameters:
  • name (str) – Human-readable name of the environment.

  • config (dict) – Free-format dictionary that contains the benchmark environment configuration. Each config must have at least the “tunable_params” and the “const_args” sections. RemoteEnv must also have at least some of the following parameters: {setup, run, teardown, wait_boot}

  • global_config (dict) – Free-format dictionary of global parameters (e.g., security credentials) to be mixed in into the “const_args” section of the local config.

  • tunables (TunableGroups) – A collection of tunable parameters for all environments.

  • service (Service) – An optional service object (e.g., providing methods to deploy or reboot a VM, etc.).

setup(tunables: mlos_bench.tunables.tunable_groups.TunableGroups, global_config: dict | None = None) bool[source]

Check if the host is up and running; boot it, if necessary.

Parameters:
  • tunables (TunableGroups) – A collection of groups of tunable parameters along with the parameters’ values. HostEnv tunables are variable parameters that, together with the HostEnv configuration, are sufficient to provision and start a host.

  • global_config (dict) – Free-format dictionary of global parameters of the environment that are not used in the optimization process.

Returns:

is_success – True if operation is successful, false otherwise.

Return type:

bool

teardown() None[source]

Clean up and shut down the host without deprovisioning it.

Return type:

None