code: def testPolicySetScalarizationParameters(self):
      agent = greedy_multi_objective_agent.GreedyMultiObjectiveNeuralAgent(
          self._time_step_spec,
          self._action_spec,
          self._scalarizer,
          objective_network_and_loss_fn_sequence=self._create_objective_network_and_loss_fn_sequence(),
          optimizer=None,
      )
      observations = tf.constant([[1, 2], [2, 1]], dtype=tf.float32)
      time_steps = ts.restart(observations, batch_size=2)
      policy = agent.policy
      policy.scalarizer.set_parameters(
          direction=tf.constant(
              [[0, 1, 0], [0, 0, 1]], dtype=tf.float32
          ),
          transform_params={
              multi_objective_scalarizer.HyperVolumeScalarizer.SLOPE_KEY: (
                  tf.constant(
                      [[0.2, 0.2, 0.2], [0.1, 0.1, 0.1]],
                      dtype=tf.float32,
                  )
              ),
              multi_objective_scalarizer.HyperVolumeScalarizer.OFFSET_KEY: (
                  tf.constant([[1, 1, 1], [2, 2, 2]], dtype=tf.float32)
              ),
          },
      )
      action_step = policy.action(time_steps)
      # Batch size 2.
      self.assertAllEqual([2], action_step.action.shape)
      self.evaluate(tf.compat.v1.initialize_all_variables())
      actions = self.evaluate(action_step.action)
      self.assertAllEqual(actions, [2, 1])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from typing import List, Tuple
  import numpy
  import tensorflow
  from tf_agents.bandits.agents import greedy_multi_objective_neural_agent
  from tf_agents.bandits.drivers import driver_utils
  from tf_agents.bandits.multi_objective import multi_objective_scalarizer
  from tf_agents.bandits.networks import global_and_arm_feature_network
  from tf_agents.bandits.networks import heteroscedastic_q_network
  from tf_agents.bandits.specs import utils
  from tf_agents.networks import network
  from tf_agents.policies import utils
  from tf_agents.specs import array_spec
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.typing import types
  from tf_agents.utils import common
label_desc: Use image node_id 12 for calling the MultiObjectiveAgentTest obj's underlying member method code with example usage: obj.testPolicySetScalarizationParameters() without return types