Merge pull request yusuketomoto#28 from SoraLab/develop

Support Python 2 & 3 simultaneously
vermapratyush · Aug 6, 2016 · fdcb730 · fdcb730
2 parents a9cdbd4 + 46965f0
commit fdcb730
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/generate.py b/generate.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 import numpy as np
 import argparse
 from PIL import Image
@@ -33,6 +34,6 @@
 result = result.transpose(0, 2, 3, 1)
 result = result.reshape((result.shape[1:]))
 result = np.uint8(result)
-print time.time() - start, 'sec'
+print(time.time() - start, 'sec')
 
 Image.fromarray(result).save(args.out)