tm.unwrap: Error when out_vectors is list [documentation]

[deasmhumhna]

import jax.numpy as jnp 
import tree_math as tm

def f(x, y):
  return x, y
  
x = y = tm.Vector(jnp.array(0.))

tm.unwrap(f, out_vectors = (True, False))(x, y)
# (tree_math.Vector(DeviceArray(0., dtype=float32, weak_type=True)), DeviceArray(0., dtype=float32, weak_type=True))
tm.unwrap(f, out_vectors = [True, False])(x, y)
# ValueError: Expected list, got (DeviceArray(0., dtype=float32, weak_type=True), DeviceArray(0., dtype=float32, weak_type=True)).

[deasmhumhna]

This is definitely just due to a lack of documentation since out_vectors has to be a pytree with the same structure as the return value.