kd.konfig.register_aliases

kauldron.konfig.register_aliases(aliases: dict[str, str]) → None

Register module aliases for nicer display.

Example:

konfig.register_aliases({
    'jax.numpy': 'jnp',
    'tensorflow.experimental.numpy': 'tnp',
})

with konfig.imports()
  import jax.numpy as jnp

assert repr(jnp.int32) == 'jnp.int32'
# Without aliases, repr(jnp.int32) == 'jax.numpy.int32'

Parameters:

aliases – The mapping import name to display alias.