code: def _zero_if_inexact_array_else_none(x):
      return 0 if eqx.is_inexact_array(x) else None
  
imports: import functools
  from typing import Any, Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 2 for calling a global function with example usage: _zero_if_inexact_array_else_none(x) and returns: unknown