snake.toolkit.cli.config

snake.toolkit.cli.config#

Configuration of SNAKE using Hydra.

Functions

cleanup_cuda

Cleanup CUDA.

conf_validator

Validate the simulation configuration.

make_hydra_cli

Create a Hydra CLI for the function.

snake_handler_resolver

Get Custom resolver for OmegaConf to get handler name.

snake_sampler_resolver

Get Custom resolver for OmegaConf to get handler name.

Classes

ConfigSNAKE

Configuration for SNAKE.

EngineConfig

Engine configuration for SNAKE.

PhantomConfig

PhantomConfig.

StatConfig

Statistical configuration for SNAKE.