aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMihai Maruseac <mihaimaruseac@google.com>2020-09-20 20:19:51 -0700
committerGitHub <noreply@github.com>2020-09-20 20:19:51 -0700
commite99e53dda53644e49f4b8b4ec16ef92f6399fc3b (patch)
tree6aa6cbdafd7af6504ccdbdf18bd46eabf2f2491d
parent7ef5c62a21f2c03483c21566dd6c048218dced26 (diff)
parentbad36df000e97cfe0a271e08778a81db4ce8834a (diff)
downloadtensorflow-e99e53dda53644e49f4b8b4ec16ef92f6399fc3b.tar.gz
Merge pull request #43410 from tensorflow/mm-fix-1.15
Fix broken tests after cherrypicks
-rw-r--r--tensorflow/python/ops/raw_ops_test.py1
-rw-r--r--tensorflow/python/ops/sparse_ops_test.py1
2 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/ops/raw_ops_test.py b/tensorflow/python/ops/raw_ops_test.py
index 0dbd7dcb916..089db57576c 100644
--- a/tensorflow/python/ops/raw_ops_test.py
+++ b/tensorflow/python/ops/raw_ops_test.py
@@ -32,7 +32,6 @@ from tensorflow.python.platform import test
@test_util.run_all_in_graph_and_eager_modes
-@test_util.disable_tfrt
class RawOpsTest(test.TestCase, parameterized.TestCase):
def testSimple(self):
diff --git a/tensorflow/python/ops/sparse_ops_test.py b/tensorflow/python/ops/sparse_ops_test.py
index 097ec66f0ae..e739b177376 100644
--- a/tensorflow/python/ops/sparse_ops_test.py
+++ b/tensorflow/python/ops/sparse_ops_test.py
@@ -24,6 +24,7 @@ import numpy as np
from tensorflow.python.eager import context
from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes
+from tensorflow.python.framework import errors
from tensorflow.python.framework import sparse_tensor
from tensorflow.python.framework import test_util
# Need array_grad to register gradient for Identity.