Skip to content
This repository has been archived by the owner on Jul 7, 2023. It is now read-only.

Commit

Permalink
Remove flaky testNoShuffleFail test that was testing non-guaranteed b…
Browse files Browse the repository at this point in the history
…ehavior

PiperOrigin-RevId: 207297174
  • Loading branch information
Ryan Sepassi authored and Copybara-Service committed Aug 3, 2018
1 parent 927f78c commit 6edbd6b
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions tensor2tensor/data_generators/problem_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,18 +73,6 @@ def testNoShufflePreprocess(self):
with tf.Session() as sess:
self.assertTrue(assert_tensors_equal(sess, tensor1, tensor2, 20))

def testNoShuffleFail(self):
problem = algorithmic.TinyAlgo()
dataset = problem.dataset(mode=tf.estimator.ModeKeys.TRAIN,
data_dir=algorithmic.TinyAlgo.data_dir,
shuffle_files=True)

tensor1 = dataset.make_one_shot_iterator().get_next()["targets"]
tensor2 = dataset.make_one_shot_iterator().get_next()["targets"]

with tf.Session() as sess:
self.assertFalse(assert_tensors_equal(sess, tensor1, tensor2, 20))


if __name__ == "__main__":
tf.test.main()

0 comments on commit 6edbd6b

Please sign in to comment.