code: def state_dict(self, *args, **kwargs):
      return {"counter": self.counter}
  
file_path: pytorch-lightning/tests/tests_data/streaming/test_combined.py
imports: import pytest
  from lightning.data.streaming.combined import CombinedStreamingDataset
  from torch.utils.data import IterableDataset
label_desc: Use image node_id 5 for calling the TestStatefulDataset obj's underlying member method code with example usage: obj.state_dict() and returns: dict