core.datasets._utils

core.datasets._utils#

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.

Functions#

rename_data_object_keys(...)

Rename data object keys

Module Contents#

core.datasets._utils.rename_data_object_keys(data_object: fairchem.core.datasets.atomic_data.AtomicData, key_mapping: dict[str, str | list[str]]) fairchem.core.datasets.atomic_data.AtomicData#

Rename data object keys

Parameters:
  • data_object – data object

  • key_mapping – dictionary specifying keys to rename and new names {prev_key: new_key}

  • keys (new_key can be a list of new)

  • example (for)

:param : :param prev_key: energy :param new_key: [common_energy, oc20_energy] :param This is currently required when we use a single target/label for multiple tasks: