Profiler patches torch without reverting it

[awaelchli]

🐛 Bug

LitData patches the worker loop in torch here

litdata/src/litdata/streaming/dataloader.py

Lines 454 to 456 in 5d6b8f9

	from torch.utils.data._utils import worker
	worker._worker_loop = _ProfileWorkerLoop(self._loader._profile_batches, self._loader._profile_dir)

but this never gets reverted. Subsequent runs with a dataloader will just happily run this worker loop. This hack should be removed, or if it stays it should have a way for it to remain disabled and be idempotent.

This could lead to hard to debug issues on the user's side.
On our side, this currently leads to interactions in our test suite.

[tchaton]

You are right !