From e9afcba6b37ab7a022af2f0b0ea3a99b1c30c060 Mon Sep 17 00:00:00 2001 From: Sean Morgan Date: Sun, 26 Apr 2020 14:39:35 -0400 Subject: [PATCH] Add testing dependency for bazel tests --- tensorflow_addons/activations/BUILD | 1 + tensorflow_addons/callbacks/BUILD | 1 + tensorflow_addons/image/BUILD | 1 + tensorflow_addons/layers/BUILD | 1 + tensorflow_addons/losses/BUILD | 1 + tensorflow_addons/optimizers/BUILD | 1 + tensorflow_addons/rnn/BUILD | 1 + tensorflow_addons/seq2seq/BUILD | 1 + tensorflow_addons/text/BUILD | 2 ++ 9 files changed, 10 insertions(+) diff --git a/tensorflow_addons/activations/BUILD b/tensorflow_addons/activations/BUILD index ae2a205b0e..deab70033e 100644 --- a/tensorflow_addons/activations/BUILD +++ b/tensorflow_addons/activations/BUILD @@ -8,6 +8,7 @@ py_library( data = [ "//tensorflow_addons:options.py", "//tensorflow_addons/custom_ops/activations:_activation_ops.so", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/callbacks/BUILD b/tensorflow_addons/callbacks/BUILD index fa4b07f319..c4ba97bbab 100644 --- a/tensorflow_addons/callbacks/BUILD +++ b/tensorflow_addons/callbacks/BUILD @@ -7,6 +7,7 @@ py_library( srcs = glob(["*.py"]), deps = [ "//tensorflow_addons/optimizers", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/image/BUILD b/tensorflow_addons/image/BUILD index 49464fb06d..92b62bad97 100644 --- a/tensorflow_addons/image/BUILD +++ b/tensorflow_addons/image/BUILD @@ -10,6 +10,7 @@ py_library( "//tensorflow_addons/custom_ops/image:_distort_image_ops.so", "//tensorflow_addons/custom_ops/image:_image_ops.so", "//tensorflow_addons/custom_ops/image:_resampler_ops.so", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/layers/BUILD b/tensorflow_addons/layers/BUILD index 64dadd4a6a..bc358b4035 100644 --- a/tensorflow_addons/layers/BUILD +++ b/tensorflow_addons/layers/BUILD @@ -10,6 +10,7 @@ py_library( ], deps = [ "//tensorflow_addons/activations", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/losses/BUILD b/tensorflow_addons/losses/BUILD index 8b7d39bf64..ec2d2aff34 100644 --- a/tensorflow_addons/losses/BUILD +++ b/tensorflow_addons/losses/BUILD @@ -7,6 +7,7 @@ py_library( srcs = glob(["*.py"]), deps = [ "//tensorflow_addons/activations", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/optimizers/BUILD b/tensorflow_addons/optimizers/BUILD index 049d8655d0..5d0f1969d1 100644 --- a/tensorflow_addons/optimizers/BUILD +++ b/tensorflow_addons/optimizers/BUILD @@ -6,6 +6,7 @@ py_library( name = "optimizers", srcs = glob(["*.py"]), deps = [ + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/rnn/BUILD b/tensorflow_addons/rnn/BUILD index 753e8d9206..6a3cc3cd09 100644 --- a/tensorflow_addons/rnn/BUILD +++ b/tensorflow_addons/rnn/BUILD @@ -6,6 +6,7 @@ py_library( name = "rnn", srcs = glob(["*.py"]), deps = [ + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/seq2seq/BUILD b/tensorflow_addons/seq2seq/BUILD index ce274d3489..9be29ee280 100644 --- a/tensorflow_addons/seq2seq/BUILD +++ b/tensorflow_addons/seq2seq/BUILD @@ -9,6 +9,7 @@ py_library( "//tensorflow_addons/custom_ops/seq2seq:_beam_search_ops.so", ], deps = [ + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], ) diff --git a/tensorflow_addons/text/BUILD b/tensorflow_addons/text/BUILD index a942037291..fe5bc04d3f 100644 --- a/tensorflow_addons/text/BUILD +++ b/tensorflow_addons/text/BUILD @@ -10,11 +10,13 @@ py_library( data = select({ "//tensorflow_addons:windows": [ "//tensorflow_addons/custom_ops/text:_skip_gram_ops.so", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], "//conditions:default": [ "//tensorflow_addons/custom_ops/text:_parse_time_op.so", "//tensorflow_addons/custom_ops/text:_skip_gram_ops.so", + "//tensorflow_addons/testing", "//tensorflow_addons/utils", ], }),