!8534 fix CSVDataset get_dataset_size interface

From: @anzhengqi
Reviewed-by: @liucunwei,@jonyguo
Signed-off-by: @liucunwei
pull/8534/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 9be483f386

@ -5489,7 +5489,7 @@ class CSVDataset(SourceDataset):
num_rows = CsvOp.get_num_rows(self.dataset_files, self.column_names is None)
self.dataset_size = get_num_rows(num_rows, self.num_shards)
if self.num_samples is not None and self.num_samples < self.dataset_size:
self.dataset_size = num_rows
self.dataset_size = self.num_samples
return self.dataset_size
def is_shuffled(self):

Loading…
Cancel
Save