code: def _test_parametric_dqn_net_builder(
      self, chooser: ParametricDQNNetBuilder__Union
  ) -> None:
      builder = chooser.value
      state_dim = 3
      state_normalization_data = NormalizationData(
          dense_normalization_parameters={
              i: NormalizationParameters(
                  feature_type=CONTINUOUS, mean=0.0, stddev=1.0
              )
              for i in range(state_dim)
          }
      )
      action_dim = 2
      action_normalization_data = NormalizationData(
          dense_normalization_parameters={
              i: NormalizationParameters(
                  feature_type=CONTINUOUS, mean=0.0, stddev=1.0
              )
              for i in range(action_dim)
          }
      )
      q_network = builder.build_q_network(
          state_normalization_data, action_normalization_data
      )
      x = q_network.input_prototype()
      y = q_network(*x)
      self.assertEqual(y.shape, (1, 1))
      serving_module = builder.build_serving_module(
          q_network, state_normalization_data, action_normalization_data
      )
      self.assertIsInstance(
          serving_module, ParametricDqnPredictorWrapper
      )
  
file_path: ReAgent/reagent/test/net_builder/test_parametric_dqn_net_builder.py
imports: import unittest
  from reagent.core.fb_checker import IS_FB_ENVIRONMENT
  from reagent.core.parameters import NormalizationData, NormalizationParameters
  from reagent.net_builder import parametric_dqn
  from reagent.net_builder.unions import ParametricDQNNetBuilder__Union
  from reagent.preprocessing.identify_types import CONTINUOUS
label_desc: Use image node_id 1 for calling the TestParametricDQNNetBuilder obj's underlying member method code with example usage: obj._test_parametric_dqn_net_builder(chooser) without return types