freeze

nyx.core.paramtree.freeze(tree, *selectors)[source]

Return tree with the Parameters at selectors frozen.

Parameters:
  • tree (pytree)

  • *selectors (str or callable) – A parameter name as parameters_table() prints it, optionally globbed, or an eqx.tree_at callable.

Return type:

pytree

Raises:

KeyError – If a pattern matches no parameter.