emmi.pipeline.batch_processors.rename_keys

Classes

RenameKeysBatchProcessor

Utility processor that simply renames the dictionary keys in a batch.

Module Contents

class emmi.pipeline.batch_processors.rename_keys.RenameKeysBatchProcessor(key_map)

Bases: ksuit.data.BatchProcessor

Utility processor that simply renames the dictionary keys in a batch.

Initializes the RenameKeysPostCollator

Parameters:

key_map (dict[str, str]) – dict with source keys as keys and target keys as values. The source keys are renamed target keys.

key_map
denormalize(key, value)

Inverts the key mapping from the __call__ method.

Parameters:

  • key (str) – The name of the item.

  • value (torch.Tensor) – The value of the item.

Returns:

The (potentially) remapped name and the unchanged value.

Return type:

(key, value)