From dd0d9fe9ab927cb55a99d6f84e60cc4149d79408 Mon Sep 17 00:00:00 2001
From: yanghaitao1 <yanghaitao1@huawei.com>
Date: Fri, 18 Sep 2020 23:41:22 -0400
Subject: [PATCH] fix ci error

---
 tests/ut/python/dataset/test_paddeddataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/ut/python/dataset/test_paddeddataset.py b/tests/ut/python/dataset/test_paddeddataset.py
index afe52f3e858..4dbc187447b 100644
--- a/tests/ut/python/dataset/test_paddeddataset.py
+++ b/tests/ut/python/dataset/test_paddeddataset.py
@@ -504,7 +504,7 @@ def test_celeba_padded():
     count = 0
     for _ in data.create_dict_iterator(num_epochs=1, output_numpy=True):
         count = count + 1
-    assert count == 4 
+    assert count == 2
 
 
 if __name__ == '__main__':