Updating stale DistributionStrategy test. (#6281)

4b566d4e · Tayo Oguntebi · Toby Boyd · e1ae37c4 · 4b566d4e
Commit 4b566d4e authored Feb 28, 2019 by Tayo Oguntebi Committed by Toby Boyd Feb 28, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 6 deletions

official/utils/misc/distribution_utils_test.py official/utils/misc/distribution_utils_test.py +0 -6

No files found.
--- a/official/utils/misc/distribution_utils_test.py
+++ b/official/utils/misc/distribution_utils_test.py
@@ -44,12 +44,6 @@ class GetDistributionStrategyTest(tf.test.TestCase):
    for device in ds.extended.worker_devices:
      self.assertIn('GPU', device)

-  def test_override_strategy(self):
-    ds = distribution_utils.get_distribution_strategy(
-        distribution_strategy='collective', num_gpus=2)
-    self.assertTrue(
-        isinstance(ds, tf.contrib.distribute.CollectiveAllReduceStrategy))
-

 class PerDeviceBatchSizeTest(tf.test.TestCase):
  """Tests for per_device_batch_size."""