Make 'collections' a list, as documented and expected by downstream custom getters.

PiperOrigin-RevId: 174184867
This commit is contained in:
A. Unique TensorFlower 2017-11-01 08:45:04 -07:00 committed by TensorFlower Gardener
parent a956486be2
commit 9da02be116

View File

@ -201,7 +201,7 @@ def variable(name, shape=None, dtype=None, initializer=None,
else [ops.GraphKeys.GLOBAL_VARIABLES])
# Remove duplicates
collections = set(collections)
collections = list(set(collections))
getter = variable_scope.get_variable
if custom_getter is not None:
getter = functools.partial(custom_getter,