Fix test that requires xla flag to be defined (#7072)

adc8f11b · Haoyu Zhang · Dong Lin · 092def7b · adc8f11b
Commit adc8f11b authored Jun 20, 2019 by Haoyu Zhang Committed by Dong Lin Jun 20, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

official/resnet/keras/keras_imagenet_test.py official/resnet/keras/keras_imagenet_test.py +1 -2

No files found.
--- a/official/resnet/keras/keras_imagenet_test.py
+++ b/official/resnet/keras/keras_imagenet_test.py
@@ -49,8 +49,7 @@ class KerasImagenetTest(googletest.TestCase):
  @classmethod
  def setUpClass(cls):  # pylint: disable=invalid-name
    super(KerasImagenetTest, cls).setUpClass()
-    imagenet_main.define_imagenet_flags()
-    keras_common.define_keras_flags()
+    keras_imagenet_main.define_imagenet_keras_flags()

  def setUp(self):
    super(KerasImagenetTest, self).setUp()