Hi @Abirate, thank you for this bug report! This is our fault - we recently changed the to_tf_dataset method to always require a collate_fn. I’m working on updating the course materials right now, and I’ll let you know as soon as a fixed version is available.