Browse Source

!8534 fix CSVDataset get_dataset_size interface

From: @anzhengqi
Reviewed-by: @liucunwei,@jonyguo
Signed-off-by: @liucunwei
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
9be483f386
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/dataset/engine/datasets.py

+ 1
- 1
mindspore/dataset/engine/datasets.py View File

@@ -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