From f3f1835aed4ab1874c0891c487cd6d0340fed67b Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Sun, 20 Sep 2020 20:16:38 -0700 Subject: [PATCH 1/2] No `disable_tfrt` present on this branch --- tensorflow/python/ops/raw_ops_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tensorflow/python/ops/raw_ops_test.py b/tensorflow/python/ops/raw_ops_test.py index 0dbd7dcb9169e7..089db57576c00e 100644 --- a/tensorflow/python/ops/raw_ops_test.py +++ b/tensorflow/python/ops/raw_ops_test.py @@ -32,7 +32,6 @@ @test_util.run_all_in_graph_and_eager_modes -@test_util.disable_tfrt class RawOpsTest(test.TestCase, parameterized.TestCase): def testSimple(self): From bad36df000e97cfe0a271e08778a81db4ce8834a Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Sun, 20 Sep 2020 20:17:26 -0700 Subject: [PATCH 2/2] Add missing import --- tensorflow/python/ops/sparse_ops_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/python/ops/sparse_ops_test.py b/tensorflow/python/ops/sparse_ops_test.py index 097ec66f0ae6af..e739b177376cac 100644 --- a/tensorflow/python/ops/sparse_ops_test.py +++ b/tensorflow/python/ops/sparse_ops_test.py @@ -24,6 +24,7 @@ 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.