Make requested review changes

diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD
index d382adb..b7c2a2d 100644 (file)
--- 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 (file)
--- 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()