From ca9651d20f36ed250b5089276f43e559ecc36468 Mon Sep 17 00:00:00 2001 From: Peter Fackeldey Date: Wed, 20 Mar 2024 14:46:35 +0100 Subject: [PATCH] update docs --- src/evermore/util.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/evermore/util.py b/src/evermore/util.py index 2f25043..278affd 100644 --- a/src/evermore/util.py +++ b/src/evermore/util.py @@ -53,11 +53,11 @@ def sum_leaves(tree: PyTree) -> Array: def tree_stack(trees: list[PyTree], broadcast_leaves: bool = False) -> PyTree: """ - Turn an array of `evm.Modifier`(s) into a `evm.Modifier` of arrays. + Turn an array of evm.Modifier(s) into a evm.Modifier of arrays. - It is important that the `jax.Array`(s) of the underlying `evm.Parameter` have the same shape. - Same applies for the effect leaves (e.g. `width`). However, the effect leaves can be - broadcasted to the same shape if `broadcast_effect_leaves` is set to `True`. + It is important that the jax.Array(s) of the underlying evm.Parameter have the same shape. + Same applies for the effect leaves (e.g. width). However, the effect leaves can be + broadcasted to the same shape if broadcast_effect_leaves is set to True. Example: