code: def setUp(self):
      super().setUp()
      self.num_neighbors = 2
      self.reward_scale = 1.0
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import exploration
label_desc: Use image node_id 1 for calling the EMIntrinsicRewardTest obj's underlying member method code with example usage: obj.setUp() without return types