c13bb7ca77d34094593c2778ee77129514aff0ee,tf_agents/networks/encoding_network_test.py,EncodingNetworkTest,testCombinedFeatureColumnInput,#EncodingNetworkTest#,216

Before Change

    specs[numeric_key] = tensor_spec.TensorSpec([state_dims], tf.int32)
    expected_dim += state_dims

    network = encoding_network.EncodingNetwork(
        specs,
        preprocessing_combiner=tf.keras.layers.DenseFeatures(columns.values()))

    output, _ = network(tensors)
    expected_shape = (batch_size, expected_dim)
    self.assertEqual(expected_shape, output.shape)

After Change

    specs[numeric_key] = tensor_spec.TensorSpec([state_dims], tf.int32)
    expected_dim += state_dims

    dense_features = tf.compat.v2.keras.layers.DenseFeatures(columns.values())
    network = encoding_network.EncodingNetwork(
        specs, preprocessing_combiner=dense_features)

    output, _ = network(tensors)
    expected_shape = (batch_size, expected_dim)
    self.assertEqual(expected_shape, output.shape)

Instances