code: def test_tree_at_none_leaf():
      with pytest.raises(ValueError):
          eqx.tree_at(lambda y: y[0], (None, None, 0), True)
      x = eqx.tree_at(
          lambda y: y[0],
          (None, None, 0),
          True,
          is_leaf=lambda y: y is None,
      )
      assert x == (True, None, 0)
  
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 5 for calling a global function with example usage: test_tree_at_none_leaf() without return types