forked from OSSInnovation/mindspore
fix ci error
This commit is contained in:
parent
d26df7cdcb
commit
dd0d9fe9ab
|
@ -504,7 +504,7 @@ def test_celeba_padded():
|
||||||
count = 0
|
count = 0
|
||||||
for _ in data.create_dict_iterator(num_epochs=1, output_numpy=True):
|
for _ in data.create_dict_iterator(num_epochs=1, output_numpy=True):
|
||||||
count = count + 1
|
count = count + 1
|
||||||
assert count == 4
|
assert count == 2
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Reference in New Issue