Remove the LightningDataset relying on un-maintained torchdata

[tchaton]

What does this PR do?

This PR removes the LightningDataset in favor of the map, optimize, StreamingDataset. In the future, we can re-investigate not using chunks but this would come in a second phase.

Fixes #<issue_number>

Before submitting

• Was this discussed/agreed via a GitHub issue? (not for typos and docs)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• Did you make sure to update the documentation with your changes? (if necessary)
• Did you write any new necessary tests? (not for typos and docs)
• Did you verify new and existing tests pass locally with your changes?
• Did you list all the breaking changes introduced by this pull request?
• Did you update the CHANGELOG? (not for typos, docs, test updates, or minor internal changes/refactors)

PR review

Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:

Reviewer checklist

• Is this pull request ready for review? (if not, please submit in draft mode)
• Check that all items from Before submitting are resolved
• Make sure the title is self-explanatory and the description concisely explains the PR
• Add labels and milestones (and optionally projects) to the PR so it can be classified

---

📚 Documentation preview 📚: https://pytorch-lightning--19019.org.readthedocs.build/en/19019/

cc @Borda

[github-actions]

⚡ Required checks status: All passing 🟢

Groups summary

🟢 lightning_data: CPU workflow

Check ID	Status
data-cpu (macOS-11, lightning, 3.10, 2.1)	success	✅
data-cpu (ubuntu-20.04, lightning, 3.10, 2.1)	success	✅
data-cpu (windows-2022, lightning, 3.10, 2.1)	success	✅

These checks are required after the changes to requirements/data/data.txt, src/lightning/data/__init__.py, src/lightning/data/backends.py, src/lightning/data/datasets/__init__.py, src/lightning/data/datasets/base.py, src/lightning/data/datasets/index.py, src/lightning/data/datasets/iterable.py, src/lightning/data/datasets/mapping.py, src/lightning/data/fileio.py, src/lightning/data/streaming/cache.py, src/lightning/data/streaming/dataloader.py, src/lightning/data/streaming/dataset.py, src/lightning/data/streaming/reader.py, src/lightning/data/streaming/shuffle.py, src/lightning/data/streaming/writer.py, src/lightning/data/utilities/env.py, tests/tests_data/datasets/__init__.py, tests/tests_data/datasets/test_data/test_index.txt, tests/tests_data/datasets/test_data/test_index_s3.txt, tests/tests_data/datasets/test_env.py, tests/tests_data/datasets/test_get_index.py, tests/tests_data/datasets/test_iterable.py, tests/tests_data/datasets/test_mapping.py, tests/tests_data/streaming/test_cache.py, tests/tests_data/streaming/test_dataset.py, tests/tests_data/test_backends.py, tests/tests_data/test_fileio.py, examples/data/image/imagenet.py.

🟢 mypy

Check ID	Status
mypy	success	✅

These checks are required after the changes to requirements/data/data.txt, src/lightning/data/__init__.py, src/lightning/data/backends.py, src/lightning/data/datasets/__init__.py, src/lightning/data/datasets/base.py, src/lightning/data/datasets/index.py, src/lightning/data/datasets/iterable.py, src/lightning/data/datasets/mapping.py, src/lightning/data/fileio.py, src/lightning/data/streaming/cache.py, src/lightning/data/streaming/dataloader.py, src/lightning/data/streaming/dataset.py, src/lightning/data/streaming/reader.py, src/lightning/data/streaming/shuffle.py, src/lightning/data/streaming/writer.py, src/lightning/data/utilities/env.py.

🟢 install

Check ID	Status
install-pkg (ubuntu-22.04, app, 3.8)	success	✅
install-pkg (ubuntu-22.04, app, 3.11)	success	✅
install-pkg (ubuntu-22.04, fabric, 3.8)	success	✅
install-pkg (ubuntu-22.04, fabric, 3.11)	success	✅
install-pkg (ubuntu-22.04, pytorch, 3.8)	success	✅
install-pkg (ubuntu-22.04, pytorch, 3.11)	success	✅
install-pkg (ubuntu-22.04, lightning, 3.8)	success	✅
install-pkg (ubuntu-22.04, lightning, 3.11)	success	✅
install-pkg (ubuntu-22.04, notset, 3.8)	success	✅
install-pkg (ubuntu-22.04, notset, 3.11)	success	✅
install-pkg (macOS-12, app, 3.8)	success	✅
install-pkg (macOS-12, app, 3.11)	success	✅
install-pkg (macOS-12, fabric, 3.8)	success	✅
install-pkg (macOS-12, fabric, 3.11)	success	✅
install-pkg (macOS-12, pytorch, 3.8)	success	✅
install-pkg (macOS-12, pytorch, 3.11)	success	✅
install-pkg (macOS-12, lightning, 3.8)	success	✅
install-pkg (macOS-12, lightning, 3.11)	success	✅
install-pkg (macOS-12, notset, 3.8)	success	✅
install-pkg (macOS-12, notset, 3.11)	success	✅
install-pkg (windows-2022, app, 3.8)	success	✅
install-pkg (windows-2022, app, 3.11)	success	✅
install-pkg (windows-2022, fabric, 3.8)	success	✅
install-pkg (windows-2022, fabric, 3.11)	success	✅
install-pkg (windows-2022, pytorch, 3.8)	success	✅
install-pkg (windows-2022, pytorch, 3.11)	success	✅
install-pkg (windows-2022, lightning, 3.8)	success	✅
install-pkg (windows-2022, lightning, 3.11)	success	✅
install-pkg (windows-2022, notset, 3.8)	success	✅
install-pkg (windows-2022, notset, 3.11)	success	✅

These checks are required after the changes to src/lightning/data/__init__.py, src/lightning/data/backends.py, src/lightning/data/datasets/__init__.py, src/lightning/data/datasets/base.py, src/lightning/data/datasets/index.py, src/lightning/data/datasets/iterable.py, src/lightning/data/datasets/mapping.py, src/lightning/data/fileio.py, src/lightning/data/streaming/cache.py, src/lightning/data/streaming/dataloader.py, src/lightning/data/streaming/dataset.py, src/lightning/data/streaming/reader.py, src/lightning/data/streaming/shuffle.py, src/lightning/data/streaming/writer.py, src/lightning/data/utilities/env.py, requirements/data/data.txt.

---

Thank you for your contribution! 💜

Note
This comment is automatically generated and updates for 60 minutes every 180 seconds. If you have any other questions, contact carmocca for help.