From ad6ccc651c6a0ab9abbb2da8a620906f11690df6 Mon Sep 17 00:00:00 2001 From: Scott Zhu Date: Mon, 8 Jun 2020 08:47:38 -0700 Subject: [PATCH] Update sequence_feature_column_test to not rely on Keras. PiperOrigin-RevId: 315282586 Change-Id: I0457ae4072aa672ae6be1bfa176b3b9f3b8fea0d --- .../python/feature_column/sequence_feature_column_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/feature_column/sequence_feature_column_test.py b/tensorflow/python/feature_column/sequence_feature_column_test.py index d0cf5ee7670..e0cd73d17e4 100644 --- a/tensorflow/python/feature_column/sequence_feature_column_test.py +++ b/tensorflow/python/feature_column/sequence_feature_column_test.py @@ -24,6 +24,7 @@ from absl.testing import parameterized import numpy as np from tensorflow.python.client import session +from tensorflow.python.feature_column import feature_column_lib as fc_lib from tensorflow.python.feature_column import feature_column_v2 as fc from tensorflow.python.feature_column import sequence_feature_column as sfc from tensorflow.python.feature_column import serialization @@ -31,7 +32,6 @@ from tensorflow.python.framework import dtypes from tensorflow.python.framework import ops from tensorflow.python.framework import sparse_tensor from tensorflow.python.framework import test_util -from tensorflow.python.keras.engine.base_layer import Layer from tensorflow.python.ops import array_ops from tensorflow.python.ops import lookup_ops from tensorflow.python.ops import math_ops @@ -132,7 +132,8 @@ def _get_sequence_dense_tensor(column, features): def _get_sequence_dense_tensor_state(column, features): - state_manager = fc._StateManagerImpl(Layer(), trainable=True) + state_manager = fc._StateManagerImpl( + fc_lib.DenseFeatures(column), trainable=True) column.create_state(state_manager) dense_tensor, lengths = column.get_sequence_dense_tensor( fc.FeatureTransformationCache(features), state_manager)