Utilities

A collection of smaller auxiliary functions and classes:

maze.utils

SimpleStatsLoggingSetup

Helper class to simplify the statistics logging setup.

clear_global_state

Resets the seed and global state to ensure that consecutive tests run under the same preconditions.

setup_logging

Setup tensorboard logging, derive the logging directory from the script name.

Timeout

Timeout class, fires a TimeoutError after the given number of seconds elapsed.

tensorboard_to_pandas

Convert the tensorboard log to a pandas DataFrame.

Process

A wrapper for multiprocessing.Process that supports exception handling and return objects.

BColors

Colored command line output formatting

maze.hydra_plugins

MazeLocalLauncher

Custom Hydra launcher distributing the jobs in separate processes on the local machine.

LauncherConfig

Hardcoded launcher configuration, linking the hydra/launcher=local override to the MazeLocalLauncher class