Tensorflow2 dataset.take() doesn't work when color_mode='grayscale'

Question

When I\'m setting up my dataset like this:

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    DATADIR,
    validation_split=0.2,
    subset=\'tra