Method set

penzai/core/selectors.py:343–353

Replaces the selected subtree(s) with a fixed replacement. Args: replacement: The pytree to replace with. Returns: A modified version of the original tree, with this replacement in place of any selected subtrees.

(self, replacement: Any)

      return new_selection.deselect()

  def set(self, replacement: Any) -> Any:
    """Replaces the selected subtree(s) with a fixed replacement.

    Args:
      replacement: The pytree to replace with.

    Returns:
      A modified version of the original tree, with this replacement in place
      of any selected subtrees.
    """
    return self.apply(lambda _: replacement)

  def get_by_path(self) -> collections.OrderedDict[KeyPath, SelectedSubtree]:
    """Retrieves the selected subtree(s) based on their path(s).