Skip to content

GitLab

Commit d22500b3 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 40617320
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/recommendation/ranking/train_test.py
View file @ d22500b3
......@@ -125,6 +125,8 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
interaction=interaction,
use_orbit=use_orbit,
strategy=strategy)
default_mode = FLAGS.mode
# Training.
FLAGS.mode = 'train'
train.main('unused_args')
......@@ -134,6 +136,7 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
# Evaluation.
FLAGS.mode = 'eval'
train.main('unused_args')
FLAGS.mode = default_mode
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