6c29a4cfa6530adb71498083a2f5b0504fc83073,tf_agents/agents/ppo/ppo_policy_test.py,PPOPolicyTest,testBuild,#PPOPolicyTest#Any#,125

Before Change

  @parameterized.named_parameters(*_test_cases("test_build"))
  def testBuild(self, network_fn):
    actor_network = tf.make_template("actor_network", network_fn)
    value_network = tf.make_template("value_network", _dummy_value_net)
    policy = ppo_policy.PPOPolicy(
        self._time_step_spec, self._action_spec, actor_network=actor_network,
        value_network=value_network)

After Change

  @parameterized.named_parameters(*_test_cases("test_build"))
  def testBuild(self, network_cls):
    actor_network = network_cls(self._action_spec)
    value_network = DummyValueNet()

    policy = ppo_policy.PPOPolicy(
        self._time_step_spec,

Instances