feat: add support files

Author: CristianoPizzamiglio

.gitignore

@@ -27,3 +27,4 @@ dist/
 
 # Visual Studio Code
 .vscode
+.vscode/settings.json

tensorflow_datasets/image_classification/pneumoniamnist/CITATIONS.bib

@@ -0,0 +1 @@
+// TODO(pneumoniamnist): BibTeX citation

tensorflow_datasets/image_classification/pneumoniamnist/TAGS.txt

@@ -0,0 +1,3 @@
+# TODO(pneumoniamnist): If your dataset downloads files, then the checksums
+# will be automatically added here when running
+# `tfds build --register_checksums`.

tensorflow_datasets/image_classification/pneumoniamnist/__init__.py

@@ -17,14 +17,14 @@ def _info(self) -> tfds.core.DatasetInfo:
     return self.dataset_info_from_configs(
         features=tfds.features.FeaturesDict({
             # These are the features of your dataset like images, labels ...
-            'image': tfds.features.Image(shape=(None, None, 3)),
-            'label': tfds.features.ClassLabel(names=['no', 'yes']),
+            'image': tfds.features.Image(shape=(28, 28, 3)),
+            'label': tfds.features.ClassLabel(names=['Normal', 'Pneumonia']),
         }),
         # If there's a common (input, target) tuple from the
         # features, specify them here. They'll be used if
         # `as_supervised=True` in `builder.as_dataset`.
         supervised_keys=('image', 'label'),  # Set to `None` to disable
-        homepage='https://dataset-homepage/',
+        homepage='https://medmnist.com//',
     )
 
   def _split_generators(self, dl_manager: tfds.download.DownloadManager):

tensorflow_datasets/image_classification/pneumoniamnist/checksums.tsv

@@ -0,0 +1,23 @@
+"""pneumoniamnist dataset."""
+
+from tensorflow_datasets.image_classification.pneumoniamnist.pneumoniamnist import pneumoniamnist_dataset_builder
+import tensorflow_datasets.public_api as tfds
+
+class PneumoniamnistTest(tfds.testing.DatasetBuilderTestCase):
+  """Tests for pneumoniamnist dataset."""
+  # TODO(pneumoniamnist):
+  DATASET_CLASS = pneumoniamnist_dataset_builder.Builder
+  SPLITS = {
+      'train': 3,  # Number of fake train example
+      'test': 1,  # Number of fake test example
+  }
+
+  # If you are calling `download/download_and_extract` with a dict, like:
+  #   dl_manager.download({'some_key': 'http://a.org/out.txt', ...})
+  # then the tests needs to provide the fake output paths relative to the
+  # fake data directory
+  # DL_EXTRACT_RESULT = {'some_key': 'output_file1.txt', ...}
+
+
+if __name__ == '__main__':
+  tfds.testing.test_main()