Fix failing test. Increasing steps seems to resolves the issue.

PiperOrigin-RevId: 404425153 Change-Id: I6e1b58be9bb13c902c663a988bcf965e688d40fb
nouiz · Oct 20, 2021 · dcacb7a · dcacb7a
1 parent 4b593da
commit dcacb7a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tensorflow/compiler/mlir/tfr/examples/mnist/BUILD b/tensorflow/compiler/mlir/tfr/examples/mnist/BUILD
@@ -66,7 +66,6 @@ distribute_py_test(
     disable_v3 = True,  # Not needed. Save some resources and test time.
     python_version = "PY3",
     tags = [
-        "manual",  # TODO(b/203175246): consistently failing right now.
         "no_cuda_asan",  # Not needed, and there were issues with timeouts.
         "no_oss",  # Avoid downloading mnist data set in oss.
         "nomultivm",  # Not needed. Save some resources and test time.

diff --git a/tensorflow/compiler/mlir/tfr/examples/mnist/mnist_train.py b/tensorflow/compiler/mlir/tfr/examples/mnist/mnist_train.py
@@ -22,7 +22,7 @@
 from tensorflow.compiler.mlir.tfr.examples.mnist import ops_defs  # pylint: disable=unused-import
 from tensorflow.python.framework import load_library
 
-flags.DEFINE_integer('train_steps', 20, 'Number of steps in training.')
+flags.DEFINE_integer('train_steps', 200, 'Number of steps in training.')
 
 _lib_dir = os.path.dirname(gen_mnist_ops.__file__)
 _lib_name = os.path.basename(gen_mnist_ops.__file__)[4:].replace('.py', '.so')