From 853d10cd7dbccee2a40fab8e659f9fb4d6a80457 Mon Sep 17 00:00:00 2001
From: anzhengqi <anzhengqi.huawei.com>
Date: Thu, 12 Nov 2020 21:44:30 +0800
Subject: [PATCH] fix CSVDataset get_dataset_size interface

---
 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 b210b5dca1..727969447a 100644
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -5488,7 +5488,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):