May it be related? : https://github.com/kedro-org/kedro/issues/696 I had a couple of issues with tensorflowdataset in the past, and ended up writing my own one but I can't remember the details.