core.models.escaip.utils.data_preprocess

Contents

core.models.escaip.utils.data_preprocess#

Functions#

data_preprocess_radius_graph(...)

Module Contents#

core.models.escaip.utils.data_preprocess.data_preprocess_radius_graph(data: fairchem.core.datasets.atomic_data.AtomicData, global_cfg: fairchem.core.models.escaip.configs.GlobalConfigs, gnn_cfg: fairchem.core.models.escaip.configs.GraphNeuralNetworksConfigs, molecular_graph_cfg: fairchem.core.models.escaip.configs.MolecularGraphConfigs) → fairchem.core.models.escaip.custom_types.GraphAttentionData#