From 94cebe4d532273d8c4e043cb78e11c7d68bee95d Mon Sep 17 00:00:00 2001
From: yanghaitao1 <yanghaitao1@huawei.com>
Date: Wed, 17 Jun 2020 03:54:54 -0400
Subject: [PATCH] fix sampler get_num_samples

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

diff --git a/mindspore/dataset/engine/samplers.py b/mindspore/dataset/engine/samplers.py
index 56ef705f60..e91dca9ce0 100644
--- a/mindspore/dataset/engine/samplers.py
+++ b/mindspore/dataset/engine/samplers.py
@@ -115,6 +115,8 @@ class Sampler:
         return self.child_sampler.is_sharded()
 
     def get_num_samples(self):
+        if self.num_samples is None:
+            return None
         return self._get_indices().size