Skip to content

GitLab

Commit 9332b71f authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Fixing order dependent test: official/recommendation/ranking/train_test.py 

PiperOrigin-RevId: 439332128
parent 7df0c6f7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/recommendation/ranking/train_test.py
View file @ 9332b71f
...@@ -125,6 +125,8 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase): ...@@ -125,6 +125,8 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
interaction=interaction, interaction=interaction,
use_orbit=use_orbit, use_orbit=use_orbit,
strategy=strategy) strategy=strategy)
default_mode = FLAGS.mode
# Training. # Training.
FLAGS.mode = 'train' FLAGS.mode = 'train'
train.main('unused_args') train.main('unused_args')
...@@ -134,6 +136,7 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase): ...@@ -134,6 +136,7 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
# Evaluation. # Evaluation.
FLAGS.mode = 'eval' FLAGS.mode = 'eval'
train.main('unused_args') train.main('unused_args')
FLAGS.mode = default_mode
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment