sample, signature_def
    )

    self.assertEmpty(feed_dict)


class RepresentativeDatasetSaverTest(test.TestCase):
  """Test cases for RepresentativeDatasetSaver."""

  def test_save_raises_error(self):
    saver = repr_dataset.RepresentativeDatasetSaver()
    repr_ds = {'serving_default': []}

    with self.assertRaisesRegex(
        NotImplementedError, 'Method "save" is not implemented.'
    ):