Fix create_meta_graph to respect an empty collection_list.
PiperOrigin-RevId: 158073112
This commit is contained in:
parent
54ccc3e5a8
commit
3d52e4cb93
@ -510,7 +510,7 @@ def create_meta_graph_def(meta_info_def=None,
|
|||||||
meta_graph_def.saver_def.MergeFrom(saver_def)
|
meta_graph_def.saver_def.MergeFrom(saver_def)
|
||||||
|
|
||||||
# Adds collection_list.
|
# Adds collection_list.
|
||||||
if collection_list:
|
if collection_list is not None:
|
||||||
clist = collection_list
|
clist = collection_list
|
||||||
else:
|
else:
|
||||||
clist = graph.get_all_collection_keys()
|
clist = graph.get_all_collection_keys()
|
||||||
|
Loading…
Reference in New Issue
Block a user