code: def state_dict(
      self, num_workers: int = 0, batch_size: int = 1
  ) -> Dict[str, Any]:
      return {
          str(dataset_idx): dataset.state_dict(
              self._num_samples_yielded[dataset_idx],
              num_workers,
              batch_size,
          )
          for dataset_idx, dataset in enumerate(self._datasets)
      }
  
file_path: pytorch-lightning/src/lightning/data/streaming/combined.py
imports: import random
  from typing import Any, Dict, Iterator, List, Optional, Sequence
  from torch.utils.data import IterableDataset
  from lightning.data.streaming.dataset import StreamingDataset
label_desc: Use image node_id 3 for calling the _CombinedDatasetIterator obj's underlying member method code with example usage: obj.state_dict(num_workers, batch_size) and returns: unknown