Week 2, Exercise 6 Machine Learning Data Lifecycle in Production

hello everyone,

i keep getting this error and after doing some check i couldn´t figure out how to solve it.

The issue is the following: ValueError: Feature traffic_volume_xf (Tensor(“Identity_9:0”, shape=(), dtype=int64)) had invalid shape () for FixedLenFeature: must have rank at least 1

i think the problem is in this line of code when i generate the .py file to transform # Create a feature that shows if the traffic volume is greater than the mean and cast to an int

    outputs[_transformed_name(_VOLUME_KEY)] = tf.cast(  
    # Use `tf.greater` to check if the traffic volume in a row is greater than the mean of the entire traffic volumn column
    tf.greater(traffic_volume, tft.mean(tf.cast(inputs[_VOLUME_KEY], tf.float32))),