Add CIFAR100 to constant file

celinejacques · celinejacques · commit 76ef62ba9702 · 2021-04-02T17:33:44.000+02:00
diff --git a/mplc/constants.py b/mplc/constants.py
@@ -49,8 +49,9 @@
 TITANIC = "titanic"
 ESC50 = "esc50"
 IMDB = 'imdb'
+CIFAR100 = "cifar100"
 # Supported datasets
-SUPPORTED_DATASETS_NAMES = [MNIST, CIFAR10, TITANIC, ESC50, IMDB]
+SUPPORTED_DATASETS_NAMES = [MNIST, CIFAR10, TITANIC, ESC50, IMDB, CIFAR100]
 
 # Number of attempts allowed before raising an error while trying to download dataset
 NUMBER_OF_DOWNLOAD_ATTEMPTS = 3
