diff options
Diffstat (limited to 'tensorflow/python/training/moving_averages_test.py')
| -rw-r--r-- | tensorflow/python/training/moving_averages_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/training/moving_averages_test.py b/tensorflow/python/training/moving_averages_test.py index 7ec6b2597e..fd9891853a 100644 --- a/tensorflow/python/training/moving_averages_test.py +++ b/tensorflow/python/training/moving_averages_test.py @@ -218,7 +218,7 @@ class ExponentialMovingAverageTest(tf.test.TestCase): self.assertDeviceEqual("/job:dev_v0", ema.average(v0).device) self.assertDeviceEqual("/job:dev_v1", ema.average(v1).device) # However, the colocation property is maintained. - self.assertEqual(["loc:@v1"], + self.assertEqual([b"loc:@v1"], ema.average(v1).op.colocation_groups()) self.assertDeviceEqual("/job:default", ema.average(tensor2).device) |