preserve_unit

Contents

preserve_unit#

tollan.utils.np.preserve_unit(f: F) F[source]#

Strip unit from first argument, reattach to result.

Parameters:

f (Callable) – Function to wrap (first arg will have units stripped/reattached)

Returns:

Wrapped function preserving units

Return type:

F