Define xrange() for Python 3

See tensorflow#2105
auzair92 · Aug 4, 2017 · 1d5dba6 · 1d5dba6
1 parent 928c889
commit 1d5dba6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rebar/rebar_train.py b/rebar/rebar_train.py
@@ -28,6 +28,12 @@
 import rebar
 import datasets
 import logger as L
+
+try:
+  xrange          # Python 2
+except NameError:
+  xrange = range  # Python 3
+
 gfile = tf.gfile
 
 tf.app.flags.DEFINE_string("working_dir", "/tmp/rebar",