From 1187411af10cff8aaea256f5fc251d38cd5740cb Mon Sep 17 00:00:00 2001
From: yanghaitao <yanghaitao1@huawei.com>
Date: Fri, 29 May 2020 15:07:27 +0800
Subject: [PATCH] a

---
 mindspore/dataset/engine/datasets.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py
index f3703b3850..f4e9679830 100644
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -2948,7 +2948,8 @@ class GeneratorDataset(MappableDataset):
 
         if rows_from_sampler is None:
             return self._dataset_size
-
+        if self._dataset_size is None:
+            return None
         return min(rows_from_sampler, self._dataset_size)
 
     # manually set dataset_size as a temporary solution.