aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/session_bundle/gc.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-07-07 01:28:05 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-07-07 02:33:27 -0700
commit79fb8db4425f9efa712eb75a11c5ae1accc88f46 (patch)
tree3b7a08c8af56a1c4931d94a70e26472341ee1e13 /tensorflow/contrib/session_bundle/gc.py
parentb0fc9af3818e3d5385edbfa9e38d66c3000ecf71 (diff)
Fix gc.largest_export_versions() so it doesn't delete the "00000000" export folder.
Change: 126785790
Diffstat (limited to 'tensorflow/contrib/session_bundle/gc.py')
-rw-r--r--tensorflow/contrib/session_bundle/gc.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/session_bundle/gc.py b/tensorflow/contrib/session_bundle/gc.py
index ad7389d96f..307101f6f1 100644
--- a/tensorflow/contrib/session_bundle/gc.py
+++ b/tensorflow/contrib/session_bundle/gc.py
@@ -88,7 +88,7 @@ def largest_export_versions(n):
def keep(paths):
heap = []
for idx, path in enumerate(paths):
- if path.export_version:
+ if path.export_version is not None:
heapq.heappush(heap, (path.export_version, idx))
keepers = [paths[i] for _, i in heapq.nlargest(n, heap)]
return sorted(keepers)