code: def store(self, val):
      try:
          self._store.store(val)
      except StoreException as e:
          try:
              okay = bool(self._store._val == val)
          except:
              raise e from None
          else:
              if not okay:
                  raise StoreException(
                      "Store occupied with not-equal value"
                  ) from None
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  from typing import Any, Callable, NamedTuple
  import weakref
  from jax._src import config
  from jax._src import core
  from jax._src import traceback_util
  from jax._src.tree_util import tree_map
  from jax._src.util import curry
label_desc: Use image node_id 2 for calling the EqualStore obj's underlying member method code with example usage: obj.store(val) without return types