From 1f9328f3a4662085b2cb948b9ed2b03d7259f78b Mon Sep 17 00:00:00 2001
From: Scott Zhu <scottzhu@google.com>
Date: Thu, 1 Oct 2020 23:50:44 -0700
Subject: [PATCH] Add missing dependency for tests.
They were implicitly imported by other rules transitively.
PiperOrigin-RevId: 334971424
Change-Id: I9e963c67a1ae7a65969b413d1588dbc0a80e670f
---
tensorflow/python/keras/feature_column/BUILD | 3 +++
tensorflow/python/keras/saving/BUILD | 2 ++
2 files changed, 5 insertions(+)
diff --git a/tensorflow/python/keras/feature_column/BUILD b/tensorflow/python/keras/feature_column/BUILD
index 17d2cb3cd52..89eaa6d1dd2 100644
--- a/tensorflow/python/keras/feature_column/BUILD
+++ b/tensorflow/python/keras/feature_column/BUILD
@@ -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",
],
)
diff --git a/tensorflow/python/keras/saving/BUILD b/tensorflow/python/keras/saving/BUILD
index 7e92defc1d8..cfd5cd2660f 100644
--- a/tensorflow/python/keras/saving/BUILD
+++ b/tensorflow/python/keras/saving/BUILD
@@ -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",