code: def get_add_transition(i):
      """For adding into RB"""
      return {
          "state": np.ones(OBS_SHAPE) * i,
          "action": int(i),
          "reward": float(2 * i),
          "extra1": float(3 * i),
      }
  
file_path: ReAgent/reagent/test/replay_memory/extra_replay_buffer_test.py
imports: import logging
  import numpy
  import numpy.testing
  import torch
  from reagent.replay_memory.circular_replay_buffer import ReplayBuffer
  from reagent.test.base.horizon_test_base import HorizonTestBase
label_desc: Use image node_id 1 for calling a global function with example usage: get_add_transition(i) and returns: dict