def test_logit_layer_name_is_logits(self):
     model = KerasModelWrapper(self.model)
     logits_name = model._get_logits_name()
     self.assertEqual(logits_name, 'l2')
 def test_logit_layer_name_is_logits(self):
     model = KerasModelWrapper(self.model)
     logits_name = model._get_logits_name()
     self.assertEqual(logits_name, 'l2')