mattertune.registry

Module Attributes

backbone_registry	Registry for backbone modules.
data_registry	Registry for data modules.

mattertune.registry.backbone_registry = Registry[mattertune.finetune.base.FinetuneModuleBaseConfig](   discriminator='name',   registered_types=6,   duplicate_tag_policy='error',   auto_rebuild=True,   tags=[     'eqV2': EqV2BackboneConfig,     'jmp': JMPBackboneConfig,     'm3gnet': M3GNetBackboneConfig,     'mace': MACEBackboneConfig,     'mattersim': MatterSimBackboneConfig,     'orb': ORBBackboneConfig,   ] )

Registry for backbone modules.

mattertune.registry.data_registry = Registry[nshconfig._src.config.Config](   discriminator='type',   registered_types=8,   duplicate_tag_policy='error',   auto_rebuild=True,   tags=[     'atoms_list': AtomsListDatasetConfig,     'db': DBDatasetConfig,     'json': JSONDatasetConfig,     'matbench': MatbenchDatasetConfig,     'mp': MPDatasetConfig,     'mptraj': MPTrajDatasetConfig,     'omat24': OMAT24DatasetConfig,     'xyz': XYZDatasetConfig,   ] )

Registry for data modules.