mlrl.testbed_slurm.extension module

Author: Michael Rapp (michael.rapp.ml@gmail.com)

Provides classes that allow configuring the functionality to run experiments via the Slurm Workload Manager.

class mlrl.testbed_slurm.extension.SlurmExtension(*dependencies: Extension)

Bases: Extension

An extension that configures the functionality to run experiments via the Slurm Workload Manager.

configure_batch_mode(_: Namespace, batch_mode: BatchMode)

See mlrl.testbed.extensions.extension.Extension.configure_batch_mode()

get_supported_modes() → set[ExperimentMode]

See mlrl.testbed.extensions.extension.Extension.get_supported_modes()