test: add tiny dataset sets

Author: begumcig

tests/data/test_datamodule.py

@@ -29,6 +29,8 @@ def iterate_dataloaders(datamodule: PrunaDataModule) -> None:
         pytest.param("AIPodcast", dict(), marks=pytest.mark.slow),
         ("ImageNet", dict(img_size=512)),
         ("TinyCIFAR10", dict(img_size=32)),
+        ("TinyImageNet", dict(img_size=224)),
+        ("TinyMNIST", dict(img_size=28)),
         pytest.param("MNIST", dict(img_size=512), marks=pytest.mark.slow),
         ("WikiText", dict(tokenizer=bert_tokenizer)),
         pytest.param("TinyWikiText", dict(tokenizer=bert_tokenizer), marks=pytest.mark.slow),