anirudhacharya commented on a change in pull request #12485:
test_ImageRecordIter_seed_augmentation flaky test fix
URL: https://github.com/apache/incubator-mxnet/pull/12485#discussion_r237218375
##########
File path: tests/python/unittest/test_io.py
##########
@@ -32,6 +33,12 @@
import sys
from common import assertRaises
import unittest
+import sys
+
+if sys.version_info >= (3,0):
+ from itertools import zip_longest
+else:
+ from itertools import izip_longest as zip_longest
Review comment:
suggestion for handling different python versions
```
try:
from itertools import izip_longest as zip_longest
except:
from itertools import zip_longest
```
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
