core.datasets.collaters.mt_collater

core.datasets.collaters.mt_collater#

Copyright (c) Meta Platforms, Inc. and affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Classes#

MTCollater

Module Contents#

class core.datasets.collaters.mt_collater.MTCollater(task_config, exclude_keys, otf_graph: bool = False)#
exclude_keys#
task_config#
dataset_task_map#
__call__(data_list: list[fairchem.core.datasets.atomic_data.AtomicData]) → fairchem.core.datasets.atomic_data.AtomicData#
data_list_collater(data_list: list[fairchem.core.datasets.atomic_data.AtomicData], dataset_task_map: dict, exclude_keys: list) → fairchem.core.datasets.atomic_data.AtomicData#
_create_dataset_task_map(config)#
_add_missing_attr(data_list, dataset_task_map)#

add missing data object attributes as inf

Contents