add get_class_indexing exception info

2020-04-14 20:42:20 +08:00 · 2020-04-14 20:42:20 +08:00 · 950d75e376
parent 8674e0ad96
commit 950d75e376
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@ -782,7 +782,7 @@ class Dataset:
        """
        if self.input:
            return self.input[0].get_class_indexing()
-        return None
+        raise NotImplementedError("Dataset {} has not supported api get_class_indexing yet.".format(type(self)))

    def reset(self):
        """Reset the dataset for next epoch"""