Fix incompatibility with --incompatible_depset_union

PiperOrigin-RevId: 248375816
This commit is contained in:
A. Unique TensorFlower 2019-05-15 11:52:58 -07:00 committed by TensorFlower Gardener
parent cfdfbe4e3d
commit 3bad08e34e

View File

@ -1688,9 +1688,9 @@ def _collect_deps_aspect_impl(target, ctx):
alldeps = depset()
if hasattr(ctx.rule.attr, "deps"):
for dep in ctx.rule.attr.deps:
alldeps = alldeps | depset([dep.label])
alldeps = depset([dep.label], transitive = [alldeps])
if hasattr(dep, "tf_collected_deps"):
alldeps = alldeps | dep.tf_collected_deps
alldeps = depset(transitive = [alldeps, dep.tf_collected_deps])
return struct(tf_collected_deps = alldeps)
collect_deps_aspect = aspect(