I am using feature-column dataset in my code, in newer version of TensorFlow 2.16.1 and later there is no keras.layers.DenseFeatures class in order to ready the input layer for the DNN. what is the alternative for that?
As I am using python 3.11.7 I couldn’t install TensorFlow 2.15 or earlier.

INPUT_COLS = [
    "pickup_longitude",
    "pickup_latitude",
    "dropoff_longitude",
    "dropoff_latitude",
    "passenger_count",
]

feature_columns = {colname: tf.feature_column.numeric_column(colname) for colname in INPUT_COLS}

# Build a keras DNN model using Sequential API
model = Sequential(
    [
        keras.layers.DenseFeatures(feature_columns=feature_columns.values()),
        keras.layers.Dense(units=32, activation="relu", name="h1"),
        keras.layers.Dense(units=8, activation="relu", name="h2"),
        keras.layers.Dense(units=1, activation="linear", name="output"),
    ]
)