ENH: don't delete the SavedModel directory

d2195d91 · remi cresson · bd756e6f · d2195d91
Commit d2195d91 authored Sep 25, 2018 by remi cresson
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 5 deletions

python/tricks.py python/tricks.py +1 -5

No files found.
--- a/python/tricks.py
+++ b/python/tricks.py
@@ -114,13 +114,9 @@ def CreateSavedModel(sess, inputs, outputs, directory):
    directory: the output path for the SavedModel
  """

-  directory += "/SavedModel"
  print("Create a SavedModel in " + directory)

-  # Delete the directory if it already exists
-  if os.path.exists(directory):
-    shutil.rmtree(directory)
-  
+      
  # Get graph
  graph = tf.get_default_graph()