Add missing dependency for tests.

They were implicitly imported by other rules transitively.

PiperOrigin-RevId: 334971424
Change-Id: I9e963c67a1ae7a65969b413d1588dbc0a80e670f
This commit is contained in:
Scott Zhu 2020-10-01 23:50:44 -07:00 committed by TensorFlower Gardener
parent cafa2ffdac
commit 1f9328f3a4
2 changed files with 5 additions and 0 deletions

View File

@ -87,6 +87,7 @@ tf_py_test(
"//tensorflow/python/eager:backprop",
"//tensorflow/python/eager:context",
"//tensorflow/python/feature_column:feature_column_v2",
"//tensorflow/python/keras",
"//tensorflow/python/keras:combinations",
],
)
@ -111,6 +112,7 @@ tf_py_test(
"//tensorflow/python/eager:backprop",
"//tensorflow/python/eager:context",
"//tensorflow/python/feature_column:feature_column_v2",
"//tensorflow/python/keras",
"//tensorflow/python/keras:combinations",
],
)
@ -169,6 +171,7 @@ py_test(
"//tensorflow/python:variables",
"//tensorflow/python/data/ops:dataset_ops",
"//tensorflow/python/feature_column:feature_column_v2",
"//tensorflow/python/keras:metrics", # Import it here since base_layer didn't import it due to circular dependency.
"//tensorflow/python/keras/layers:recurrent",
],
)

View File

@ -53,11 +53,13 @@ py_library(
"//tensorflow/python:tensor_spec",
"//tensorflow/python/eager:def_function",
"//tensorflow/python/keras:backend",
"//tensorflow/python/keras:losses",
"//tensorflow/python/keras:optimizers",
"//tensorflow/python/keras:regularizers",
"//tensorflow/python/keras/engine:input_spec",
"//tensorflow/python/keras/mixed_precision/experimental:autocast_variable",
"//tensorflow/python/keras/utils:engine_utils",
"//tensorflow/python/keras/utils:metrics_utils",
"//tensorflow/python/keras/utils:mode_keys",
"//tensorflow/python/saved_model",
"//tensorflow/python/saved_model/model_utils",