Updating stale DistributionStrategy test. (#6281)

tayo · tfboyd · commit 4b566d4e800f · 2019-02-28T07:46:32.000-08:00
diff --git a/official/utils/misc/distribution_utils_test.py b/official/utils/misc/distribution_utils_test.py
@@ -44,12 +44,6 @@ def test_mirrored_strategy(self):
     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."""
