Rm xrange usage from metrics_hook_test

PiperOrigin-RevId: 179882966

Author: Ryan Sepassi

tensor2tensor/utils/metrics_hook_test.py

@@ -74,14 +74,14 @@ def testStop(self):
     ckpt_dir = self.ckpt_dir("stop")
     dummy = DummyHook(ckpt_dir, every_n_steps=10)
     with self.sess(dummy, ckpt_dir) as sess:
-      for _ in xrange(20):
+      for _ in range(20):
         sess.run(incr_global_step)
 
       # Summary files should now have 2 global step values in them
       self.flush()
 
       # Run for 10 more so that the hook gets triggered again
-      for _ in xrange(10):
+      for _ in range(10):
         sess.run(incr_global_step)
 
       # Check that the metrics have actually been collected.
@@ -93,7 +93,7 @@ def testStop(self):
       self.assertTrue(len(steps) >= 2)
 
       # Run for 10 more so that the hook triggers stoppage
-      for _ in xrange(10):
+      for _ in range(10):
         sess.run(incr_global_step)
 
       with self.assertRaisesRegexp(RuntimeError, "after should_stop requested"):
@@ -117,21 +117,21 @@ def testEarlyStoppingHook(self):
         plateau_decrease=False,
         every_n_steps=10)
     with self.sess(stop_hook, ckpt_dir) as sess:
-      for _ in xrange(20):
+      for _ in range(20):
         sess.run((incr_global_step, incr_counter))
 
       # Summary files should now have 2 values in them
       self.flush()
 
       # Run for more steps so that the hook gets triggered and we verify that we
       # don't stop.
-      for _ in xrange(30):
+      for _ in range(30):
         sess.run((incr_global_step, incr_counter))
 
       self.flush()
 
       # Run without incrementing the counter
-      for _ in xrange(40):
+      for _ in range(40):
         sess.run(incr_global_step)
 
       # Metrics should be written such that now the counter has gone >20 steps
@@ -140,7 +140,7 @@ def testEarlyStoppingHook(self):
 
       # Check that we ask for stop
       with self.assertRaisesRegexp(RuntimeError, "after should_stop requested"):
-        for _ in xrange(30):
+        for _ in range(30):
           sess.run(incr_global_step)
 
   def testPlateauOpHook(self):
@@ -164,21 +164,21 @@ def testPlateauOpHook(self):
         plateau_decrease=False,
         every_n_steps=10)
     with self.sess(stop_hook, ckpt_dir) as sess:
-      for _ in xrange(20):
+      for _ in range(20):
         sess.run((incr_global_step, incr_counter))
 
       # Summary files should now have 2 values in them
       self.flush()
 
       # Run for more steps so that the hook gets triggered and we verify that we
       # don't stop.
-      for _ in xrange(30):
+      for _ in range(30):
         sess.run((incr_global_step, incr_counter))
 
       self.flush()
 
       # Run without incrementing the counter
-      for _ in xrange(30):
+      for _ in range(30):
         sess.run(incr_global_step)
       self.flush()
 
@@ -187,8 +187,8 @@ def testPlateauOpHook(self):
       # Metrics should be written such that now the counter has gone >20 steps
       # without being incremented.
       # Check that we run the incr_indicator op several times
-      for _ in xrange(3):
-        for _ in xrange(10):
+      for _ in range(3):
+        for _ in range(10):
           sess.run(incr_global_step)
         self.flush()