Skip to content

Commit

Permalink
[Enhance]bazel build python2, fix cycle import
Browse files Browse the repository at this point in the history
  • Loading branch information
gaolei7 committed Mar 16, 2022
1 parent b73e41b commit 0256cbb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow/python/training/saving/saveable_object_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@
from tensorflow.python.ops import state_ops
from tensorflow.python.ops import variables
from tensorflow.python.training.saving import saveable_object
from tensorflow.python.training import saver
from tensorflow.python.training.tracking import base as trackable
from tensorflow.python.util import object_identity

Expand Down Expand Up @@ -468,6 +467,7 @@ def validate_and_slice_inputs(names_to_saveables):
key=lambda x: x[0]):
from tensorflow.python.ops import hash_table
if isinstance(name, hash_table.HashTable):
from tensorflow.python.training import saver
_add_saveable(saveables, seen_ops, saver.HashTableSaveable(op[1], op[0]))
else:
for converted_saveable_object in saveable_objects_for_op(op, name):
Expand Down

0 comments on commit 0256cbb

Please sign in to comment.