WARNING:tensorflow:Keras is training/fitting/evaluating on array-like data

WARNING:tensorflow:Keras is training/fitting/evaluating on array-like data. Keras may not be optimized for this format, so if your input data format is supported by TensorFlow I/O (GitHub - tensorflow/io: Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO) we recommend using that to load a Dataset instead.
Epoch 1/10
WARNING:tensorflow:AutoGraph could not transform <function Model.make_train_function..train_function at 0x000001BF037591F8> and will run it as-is.
Cause: closure mismatch, requested (‘self’, ‘step_function’), but source function had ()
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert

Please provide standalone code to reproduce the issue? Thank you