Fix flakiness on windows tests.

PiperOrigin-RevId: 296513528
Change-Id: Ic365c59467b610b0f28bd96ec35da11024f93f7c
This commit is contained in:
Dero Gharibian 2020-02-21 14:34:57 -08:00 committed by TensorFlower Gardener
parent f28581b01a
commit 36b569fcd5

View File

@ -64,8 +64,8 @@ class PreprocessingStageTest(
self.assertEqual(stage.layers[0].adapt_count, 1)
self.assertEqual(stage.layers[1].adapt_count, 1)
self.assertEqual(stage.layers[2].adapt_count, 1)
self.assertLess(stage.layers[0].adapt_time, stage.layers[1].adapt_time)
self.assertLess(stage.layers[1].adapt_time, stage.layers[2].adapt_time)
self.assertLessEqual(stage.layers[0].adapt_time, stage.layers[1].adapt_time)
self.assertLessEqual(stage.layers[1].adapt_time, stage.layers[2].adapt_time)
# Check call
y = stage(array_ops.ones((3, 4)))