code: def __init__(self, size, step):
      self.size = size
      self.step = step
      self.counter = 0
  
file_path: pytorch-lightning/tests/tests_data/streaming/test_dataloader.py
imports: import torch
  from lightning.data.streaming import CombinedStreamingDataset, StreamingDataLoader
  from torch import tensor
label_desc: Use image node_id 1 to create a new TestStatefulDataset object  with example: obj = TestStatefulDataset(size, step)