fix another test

apache · Jun 23, 2018 · 2ab45a7 · 2ab45a7
1 parent 209414a
commit 2ab45a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python/unittest/test_io.py b/tests/python/unittest/test_io.py
@@ -407,7 +407,8 @@ def test_ImageRecordIter_seed_augmentation():
         mean_img="data/cifar/cifar10_mean.bin",
         shuffle=False,
         data_shape=(3, 28, 28),
-        batch_size=3)
+        batch_size=3,
+        seed_aug=seed_aug)
     batch = dataiter.next()
     data = batch.data[0].asnumpy().astype(np.uint8)