tf.data.Dataset.from_generator not working as I would

Question

I\'m pretty new to tensorflow and I\'m stuck with a problem:

I\'m trying to create a dataset from my custom data by using:

 tf.data.Dataset.from_genera