9cd26e0a175be1333bae77720b7806d62857429c,tf_agents/agents/behavioral_cloning/behavioral_cloning_agent_test.py,BehavioralCloningAgentTest,testCreateAgentDimChecks,#BehavioralCloningAgentTest#,127

Before Change

        loss_fn=lambda logits, actions: 0)

  def testCreateAgentDimChecks(self):
    action_spec = [tensor_spec.BoundedTensorSpec([1, 2], tf.int32, 0, 1)]
    cloning_net = DummyNet(self._observation_spec, action_spec)
    with self.assertRaisesRegexp(NotImplementedError, ".*one dimensional.*"):
      behavioral_cloning_agent.BehavioralCloningAgent(
          self._time_step_spec, action_spec,

After Change

          optimizer=None)

  def testCreateAgentDimChecks(self):
    action_spec = tensor_spec.BoundedTensorSpec([1, 2], tf.int32, 0, 1)
    cloning_net = DummyNet(self._observation_spec, action_spec)
    with self.assertRaisesRegexp(NotImplementedError, ".*scalar, unnested.*"):
      behavioral_cloning_agent.BehavioralCloningAgent(
          self._time_step_spec, action_spec,

Instances