forked from mindspore-Ecosystem/mindspore
!318 Add get_class_indexing exception
Merge pull request !318 from pengyanjun/fix_get_cladd_indexing_warning
This commit is contained in:
commit
9a781025ec
|
@ -827,7 +827,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"""
|
||||
|
|
Loading…
Reference in New Issue