Keras model.fit() returns TypeError: 'NoneType' object is not callable when using tf.data.Dataset.from_generator

Question

I am trying to use tensorflow data api to replace my current data loading pipeline.

X_train shape: 385880x105624 sparse matrix of type \'