Dataset object has no attribute `to_tf_dataset`

[lewtun]

Hey @rajkumar I believe the to_tf_dataset() method was only added in a recent version of datasets. Could you try upgrading to the latest version and check if the problem persists?