Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "b17e67df36e7fbb8e5e480917bb99357d8c0ec65"
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
Hide 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): ...@@ -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