# Path: model_card_toolkit/documentation/examples/cats_vs_dogs.py
# IMAGE_SIZE = 150
# BATCH_SIZE = 32
# NUM_BATCHES = 10
# DEFAULT_TRAINING_EPOCHS = 4
# def get_data() -> Dict[Text, Any]:
# def create_model(
#     training_epochs: int = DEFAULT_TRAINING_EPOCHS) -> tf.keras.Model:


from absl.testing import absltest
from model_card_toolkit.documentation.examples import cats_vs_dogs
"""Tests for model_card_toolkit.documentation.examples.cats_vs_dogs."""


SLICES = ['combined', 'cat', 'dog']
FIELDS = ['examples', 'labels']


class CatsVsDogsTest(absltest.TestCase):

  def test_get_data(self):

# Path: model_card_toolkit/documentation/examples/cats_vs_dogs.py
# IMAGE_SIZE = 150
# BATCH_SIZE = 32
# NUM_BATCHES = 10
# DEFAULT_TRAINING_EPOCHS = 4
# def get_data() -> Dict[Text, Any]:
# def create_model(
#     training_epochs: int = DEFAULT_TRAINING_EPOCHS) -> tf.keras.Model:


# Path: model_card_toolkit/documentation/examples/cats_vs_dogs_test.py
from absl.testing import absltest
from model_card_toolkit.documentation.examples import cats_vs_dogs

"""Tests for model_card_toolkit.documentation.examples.cats_vs_dogs."""


SLICES = ['combined', 'cat', 'dog']
FIELDS = ['examples', 'labels']


class CatsVsDogsTest(absltest.TestCase):

  def test_get_data(self):

    data = cats_vs_dogs.get_data()