Fixing / disabling a few tests failing when we run with TF v2. This change is

mostly the same as
#31386/commits/4de05d7afdce730c3fee0ec4aa031caa52fa82c2
but the latter also removes contrib from pip_smoke_test which has already been
done.

PiperOrigin-RevId: 269683160
This commit is contained in:
Anna R 2019-09-17 16:41:05 -07:00 committed by TensorFlower Gardener
parent d44f3a37ec
commit 290f1571a5
7 changed files with 14 additions and 7 deletions

View File

@ -27,7 +27,7 @@ class FactTest(tf.test.TestCase):
@test_util.run_deprecated_v1
def test(self):
with self.cached_session():
print(tf.user_ops.my_fact().eval())
print(tf.compat.v1.user_ops.my_fact().eval())
if __name__ == '__main__':

View File

@ -48,7 +48,8 @@ class ZeroOut1Test(tf.test.TestCase):
def testLoadTwice(self):
zero_out_loaded_again = tf.load_op_library(os.path.join(
tf.resource_loader.get_data_files_path(), 'zero_out_op_kernel_1.so'))
tf.compat.v1.resource_loader.get_data_files_path(),
'zero_out_op_kernel_1.so'))
self.assertEqual(zero_out_loaded_again, zero_out_op_1._zero_out_module)

View File

@ -47,7 +47,7 @@ class ZeroOut2Test(tf.test.TestCase):
shape = (5,)
x = tf.constant([5, 4, 3, 2, 1], dtype=tf.float32)
y = zero_out_op_2.zero_out(x)
err = tf.test.compute_gradient_error(x, shape, y, shape)
err = tf.compat.v1.test.compute_gradient_error(x, shape, y, shape)
self.assertLess(err, 1e-4)
@test_util.run_deprecated_v1
@ -56,7 +56,7 @@ class ZeroOut2Test(tf.test.TestCase):
shape = (2, 3)
x = tf.constant([[6, 5, 4], [3, 2, 1]], dtype=tf.float32)
y = zero_out_op_2.zero_out(x)
err = tf.test.compute_gradient_error(x, shape, y, shape)
err = tf.compat.v1.test.compute_gradient_error(x, shape, y, shape)
self.assertLess(err, 1e-4)

View File

@ -22,7 +22,7 @@ import os.path
import tensorflow as tf
_zero_out_module = tf.load_op_library(
os.path.join(tf.resource_loader.get_data_files_path(),
os.path.join(tf.compat.v1.resource_loader.get_data_files_path(),
'zero_out_op_kernel_1.so'))
zero_out = _zero_out_module.zero_out
namespace_zero_out = _zero_out_module.namespace_zero_out

View File

@ -22,7 +22,7 @@ import os.path
import tensorflow as tf
_zero_out_module = tf.load_op_library(
os.path.join(tf.resource_loader.get_data_files_path(),
os.path.join(tf.compat.v1.resource_loader.get_data_files_path(),
'zero_out_op_kernel_2.so'))
zero_out = _zero_out_module.zero_out
zero_out2 = _zero_out_module.zero_out2

View File

@ -22,6 +22,6 @@ import os.path
import tensorflow as tf
_zero_out_module = tf.load_op_library(
os.path.join(tf.resource_loader.get_data_files_path(),
os.path.join(tf.compat.v1.resource_loader.get_data_files_path(),
'zero_out_op_kernel_3.so'))
zero_out = _zero_out_module.zero_out

View File

@ -62,6 +62,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)
@ -98,6 +99,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)
@ -144,6 +146,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)
@ -187,6 +190,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)
@ -230,6 +234,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)
@ -284,6 +289,7 @@ tf_py_test(
],
tags = [
"no_pip", # b/131330719
"v1only", # uses contrib
],
)