I was using from ray.data.datasource.tfrecords_datasource import TFRecordDatasource in ray 2.23.0. But after upgrading to 2.40.0, this class was moved to ray.data._internal.datasource.tfrecords_datasource.TFRecordDatasource from this change.
similar to What's the migration path for ray.data.aggregate's Max, Mean, Min, and Std functions?