From 950d75e3761762541c0c8947c1ebcb3ce7ca243d Mon Sep 17 00:00:00 2001 From: Yanjun Peng Date: Tue, 14 Apr 2020 20:42:20 +0800 Subject: [PATCH] add get_class_indexing exception info --- mindspore/dataset/engine/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index f92f6f28a83..819013031fb 100644 --- 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"""