message: fixit for basic_loops_test
diff: @@ -23,7 +23,6 @@ import shutil
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  -from tensorflow.python.framework import test_util
  from tensorflow.python.platform import test
  from tensorflow.python.training import basic_loops
  from tensorflow.python.training import supervisor
  @@ -38,10 +37,8 @@ def _test_dir(test_name):
  class BasicTrainLoopTest(test.TestCase):
  - @test_util.run_deprecated_v1
  def testBasicTrainLoop(self):
  logdir = _test_dir("basic_train_loop")
  - sv = supervisor.Supervisor(logdir=logdir)
  # Counts the number of calls.
  num_calls = [0]
  @@ -53,14 +50,13 @@ class BasicTrainLoopTest(test.TestCase):
  sv.request_stop()
  with ops.Graph().as_default():
  + sv = supervisor.Supervisor(logdir=logdir)
  basic_loops.basic_train_loop(
  sv, train_fn, args=(sv, "y"), kwargs={"a": "A"})
  self.assertEqual(3, num_calls[0])
  - @test_util.run_deprecated_v1
  def testBasicTrainLoopExceptionAborts(self):
  logdir = _test_dir("basic_train_loop_exception_aborts")
  - sv = supervisor.Supervisor(logdir=logdir)
  def train_fn(unused_sess):
  train_fn.counter += 1
  @@ -71,13 +67,12 @@ class BasicTrainLoopTest(test.TestCase):
  train_fn.counter = 0
  with ops.Graph().as_default():
  + sv = supervisor.Supervisor(logdir=logdir)
  with self.assertRaisesRegex(RuntimeError, "Failed"):
  basic_loops.basic_train_loop(sv, train_fn)
  - @test_util.run_deprecated_v1
  def testBasicTrainLoopRetryOnAborted(self):
  logdir = _test_dir("basic_train_loop_exception_aborts")
  - sv = supervisor.Supervisor(logdir=logdir)
  class AbortAndRetry(object):
  @@ -95,6 +90,7 @@ class BasicTrainLoopTest(test.TestCase):
  raise RuntimeError("Failed Again")
  with ops.Graph().as_default():
  + sv = supervisor.Supervisor(logdir=logdir)
  aar = AbortAndRetry()
  with self.assertRaisesRegex(RuntimeError, "Failed Again"):
  basic_loops.basic_train_loop(sv, aar.train_fn)
  