logging

jaxns.internals.logging

Module Contents

logger

get_grandparent_info(relative_depth=7)

Get the file, line number and function name of the caller of the caller of this function.

Parameters:

• relative_depth (int) – the number of frames to go back from the caller of this function. Default is 6. Should be

• call. (enough to get out of a jax.tree.map)

Returns:

a string with the file, line number and function name of the caller of the caller of this function.

Return type:

str