* Add support for IterableDatasets everywhere * Added type hints, simplified code and improved coverage in data_loading.py * Update CHANGELOG.md
* split trainer tests * Apply suggestions from code review * format string * add CI timeout