code: def check_save_load(
      self, model: Union[DuelingQNetwork, ParametricDuelingQNetwork]
  ):
      if isinstance(model, ParametricDuelingQNetwork):
          script_model = ParametricDuelingQNetworkTorchScriptWrapper(
              model
          )
      else:
          script_model = DuelingQNetworkTorchScriptWrapper(model)
      run_model_jit_trace(model, script_model)
  
imports: import logging
  import unittest
  from typing import Union
  import torch
  import torch.nn
  from reagent.core import types
  from reagent.models.dueling_q_network import DuelingQNetwork, ParametricDuelingQNetwork
  from reagent.test.models.test_utils import run_model_jit_trace
label_desc: Use image node_id 1 for calling the TestDuelingQNetwork obj's underlying member method code with example usage: obj.check_save_load(model) without return types