code: def test_from_tfds(self):
      # TODO: Remove this once tfds download error is fixed.
      self.skipTest(
          "Temporarily skip the unittest due to tfds download error."
      )
      (
          train_data,
          validation_data,
          test_data,
      ) = dataset.Dataset.from_tfds("beans")
      self.assertIsInstance(
          train_data.gen_tf_dataset(), tf.data.Dataset
      )
      self.assertLen(train_data, 1034)
      self.assertEqual(train_data.num_classes, 3)
      self.assertEqual(
          train_data.label_names,
          ["angular_leaf_spot", "bean_rust", "healthy"],
      )
  
      self.assertIsInstance(
          validation_data.gen_tf_dataset(), tf.data.Dataset
      )
      self.assertLen(validation_data, 133)
      self.assertEqual(validation_data.num_classes, 3)
      self.assertEqual(
          validation_data.label_names,
          ["angular_leaf_spot", "bean_rust", "healthy"],
      )
  
      self.assertIsInstance(test_data.gen_tf_dataset(), tf.data.Dataset)
      self.assertLen(test_data, 128)
      self.assertEqual(test_data.num_classes, 3)
      self.assertEqual(
          test_data.label_names,
          ["angular_leaf_spot", "bean_rust", "healthy"],
      )
  
file_path: mediapipe/mediapipe/model_maker/python/vision/image_classifier/dataset_test.py
imports: import os
  import random
  import numpy
  import tensorflow
  from mediapipe.model_maker.python.vision.core import image_utils
  from mediapipe.model_maker.python.vision.core import test_utils
  from mediapipe.model_maker.python.vision.image_classifier import dataset
label_desc: Use image node_id 4 for calling the DatasetTest obj's underlying member method code with example usage: obj.test_from_tfds() without return types