code: def test_module(getkey):
      lin = eqx.nn.Linear(2, 2, key=getkey())
      mlp = eqx.nn.MLP(2, 2, 2, 2, key=getkey())
      eqx.tree_pformat(lin)
      eqx.tree_pformat(mlp)
  
file_path: equinox/tests/test_pformat.py
imports: import functools
  import typing
  import equinox
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: test_module(getkey) without return types