From 30331b3fc02d9ae259e1241b40b945d242924376 Mon Sep 17 00:00:00 2001 From: Sami Kama Date: Tue, 17 Apr 2018 15:55:27 -0700 Subject: [PATCH] Make requested review changes --- tensorflow/contrib/tensorrt/BUILD | 5 +++++ tensorflow/contrib/tensorrt/test/tf_trt_integration_test.py | 5 +++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD index d382adb..b7c2a2d 100644 --- a/tensorflow/contrib/tensorrt/BUILD +++ b/tensorflow/contrib/tensorrt/BUILD @@ -19,6 +19,7 @@ load( "tf_gen_op_libs", "tf_gen_op_wrapper_py", ) +load("//tensorflow:tensorflow.bzl", "py_test") load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library") load("//tensorflow:tensorflow.bzl", "tf_py_wrap_cc") @@ -280,6 +281,10 @@ py_test( srcs = ["test/tf_trt_integration_test.py"], main = "test/tf_trt_integration_test.py", srcs_version = "PY2AND3", + tags = [ + "manual", + "notap", + ], deps = [ ":init_py", "//tensorflow/python:client_testlib", diff --git a/tensorflow/contrib/tensorrt/test/tf_trt_integration_test.py b/tensorflow/contrib/tensorrt/test/tf_trt_integration_test.py index b17fdd5..7a47328 100644 --- a/tensorflow/contrib/tensorrt/test/tf_trt_integration_test.py +++ b/tensorflow/contrib/tensorrt/test/tf_trt_integration_test.py @@ -37,10 +37,11 @@ from tensorflow.python.platform import googletest @test_util.with_c_api class IntegrationTest(test_util.TensorFlowTestCase): """Class to test Tensorflow-TensorRT integration.""" + def setUp(self): """Setup method.""" super(IntegrationTest, self).setUp() - warnings.simplefilter('always') + warnings.simplefilter("always") inp_dims = (100, 24, 24, 2) self._input = np.random.random_sample(inp_dims) self._original_graph = self.get_simple_graph_def() @@ -151,5 +152,5 @@ class IntegrationTest(test_util.TensorFlowTestCase): self.assertAllEqual(result1, result) -if __name__ == '__main__': +if __name__ == "__main__": googletest.main() -- 2.7.4