code: def setUp(self):
      super().setUp()
      jax.config.update("jax_numpy_rank_promotion", "raise")
  
file_path: dm-haiku/haiku/_src/integration/rank_promotion_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import haiku
  from haiku._src import test_utils
  from haiku._src.integration import descriptors
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the RankPromotionTest obj's underlying member method code with example usage: obj.setUp() without return types