code: def test_tree_at_dependent_where(getkey):
      mlp = eqx.nn.MLP(2, 2, 2, 2, key=getkey())
  
      def where(m):
          return jtu.tree_leaves(eqx.filter(m, eqx.is_array))
  
      with pytest.raises(ValueError):
          eqx.tree_at(where, mlp, where(mlp))
  
imports: import equinox
  import jax
  import jax.core
  import jax.nn
  import jax.numpy
  import jax.random
  import jax.tree_util
  import numpy
  import pytest
label_desc: Use image node_id 4 for calling a global function with example usage: test_tree_at_dependent_where(getkey) and returns: jtu