Revert the change to gen_tftrt_model.py

pooyadavoodi · Dec 11, 2019 · 8a03f97 · 8a03f97
1 parent d50a54a
commit 8a03f97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/python/compiler/tensorrt/test/testdata/gen_tftrt_model.py b/tensorflow/python/compiler/tensorrt/test/testdata/gen_tftrt_model.py
@@ -45,6 +45,7 @@
 from tensorflow.python.saved_model import utils
 from tensorflow.python.training.tracking import tracking
 
+
 def GetGraph(input1, input2, var):
   """Define graph."""
   add = input1 + var
@@ -89,7 +90,6 @@ def run(self, input1, input2):
 def GenerateModelV1(tf_saved_model_dir, tftrt_saved_model_dir):
   """Generate and convert a model using TFv1 API."""
 
-
   def SimpleModel():
     """Define model with a TF graph."""
 
@@ -134,4 +134,4 @@ def GraphFn():
 if __name__ == "__main__":
   GenerateModelV2(
       tf_saved_model_dir="tf_saved_model",
-      tftrt_saved_model_dir="tftrt_saved_model")
+      tftrt_saved_model_dir="tftrt_saved_model")