How do you send arguments to a generator function using tf.data.Dataset.from_generator()?

Question

I would like to create a number of tf.data.Dataset using the from_generator() function. I would like to send an argument to the generator function (

Answer 1

For Tensorflow 2.4:

training_dataset = tf.data.Dataset.from_generator(
     raw_data_gen, 
     args=(1), 
     output_types=(tf.float32, tf.uint8), 
     output_shapes=([None, 1], [None]))