mlrl.testbed_slurm.runner module

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

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

class mlrl.testbed_slurm.runner.SlurmRunner

Bases: Runner

A BatchMode.Runner that allows to run experiments via the Slurm Workload manager.

class ConfigFile(file_path: str, schema_file_path: str)

Bases: object

A YAML configuration file that configures Slurm jobs to be run.

property after_script: List[str]

The shell commands to be executed after the experiment has finished.

property before_script: List[str]

The shell commands to be executed before the experiment is started.

property sbatch_arguments: List[str]

The sbatch arguments (starting with #SBATCH) to be passed to a Slurm job.

run_batch(args: Namespace, batch: List[Command], _: Recipe)

See mlrl.testbed.modes.mode_batch.BatchMode.Runner.run_batch()