code: def __init__(
      self,
      model: Union[
          DuelingQNetwork,
          ParametricDuelingQNetwork,
      ],
  ):
      super().__init__()
      self.model = 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 to create a new ParametricDuelingQNetworkTorchScriptWrapper object from inherited base classes: nn with example: obj = ParametricDuelingQNetworkTorchScriptWrapper(model)