From 93085639f1c9d1538afa040ac0ec1fb6c9801568 Mon Sep 17 00:00:00 2001
From: ml-illustrated <gerald@ml-illustrated.com>
Date: Fri, 3 Jan 2020 11:21:01 -0800
Subject: [PATCH] add backward compat properties to CIFAR100 dataset

---
 dataset/cifar100.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/dataset/cifar100.py b/dataset/cifar100.py
index 7f59aafe..749ad470 100644
--- a/dataset/cifar100.py
+++ b/dataset/cifar100.py
@@ -35,8 +35,29 @@ def get_data_folder():
 
     return data_folder
 
+class CIFAR100BackCompat(datasets.CIFAR100):
+    """
+    CIFAR100Instance+Sample Dataset
+    """
+
+    @property
+    def train_labels(self):
+        return self.targets
+
+    @property
+    def test_labels(self):
+        return self.targets
+
+    @property
+    def train_data(self):
+        return self.data
+
+    @property
+    def test_data(self):
+        return self.data
+    
 
-class CIFAR100Instance(datasets.CIFAR100):
+class CIFAR100Instance(CIFAR100BackCompat):
     """CIFAR100Instance Dataset.
     """
     def __getitem__(self, index):
@@ -106,7 +127,7 @@ def get_cifar100_dataloaders(batch_size=128, num_workers=8, is_instance=False):
         return train_loader, test_loader
 
 
-class CIFAR100InstanceSample(datasets.CIFAR100):
+class CIFAR100InstanceSample(CIFAR100BackCompat):
     """
     CIFAR100Instance+Sample Dataset
     """