MolecularDiffusion.modules.models.tabasco.utils.rich_utils

Attributes

log

Functions

enforce_tags(→ None)

Prompts user to input tags from command line if no tags are provided in config.

print_config_tree(, resolve, save_to_file)

Prints the contents of a DictConfig as a tree structure using the Rich library.

Module Contents

MolecularDiffusion.modules.models.tabasco.utils.rich_utils.enforce_tags(cfg: omegaconf.DictConfig, save_to_file: bool = False) → None

Prompts user to input tags from command line if no tags are provided in config.

Parameters:

  • cfg – Composed Hydra configuration.

  • save_to_file – If True, save the final tags list to cfg.paths.output_dir / ‘tags.log’.

MolecularDiffusion.modules.models.tabasco.utils.rich_utils.print_config_tree(cfg: omegaconf.DictConfig, print_order: Sequence[str] = ('data', 'model', 'callbacks', 'logger', 'trainer', 'paths', 'extras'), resolve: bool = False, save_to_file: bool = False) → None

Prints the contents of a DictConfig as a tree structure using the Rich library.

Parameters:

  • cfg – Composed Hydra configuration.

  • print_order – Sequence that controls the order of top-level keys.

  • resolve – If True, resolve ${} references before printing.

  • save_to_file – If True, also write the tree to cfg.paths.output_dir / ‘config_tree.log’.

MolecularDiffusion.modules.models.tabasco.utils.rich_utils.log