code: def test_eval_shape(getkey, tmp_path):
      model = eqx.nn.MLP(2, 2, 2, 2, key=getkey())
      eqx.tree_serialise_leaves(tmp_path, model)
  
      model2 = eqx.filter_eval_shape(
          eqx.nn.MLP, 2, 2, 2, 2, key=getkey()
      )
      model3 = eqx.tree_deserialise_leaves(tmp_path, model2)
  
      assert eqx.tree_equal(model, model3, typematch=True)
  
file_path: equinox/tests/test_serialisation.py
imports: import os
  import equinox
  import jax
  import jax.numpy
  import numpy
  import pytest
  from jax.dtypes import bfloat16
label_desc: Use image node_id 9 for calling a global function with example usage: test_eval_shape(getkey, tmp_path) without return types