code: def generate_data(num_samples):
      shapes = {"aaa": (), "bbb": (4,), "ccc": (3, 2), "ddd": (5, 1)}
      return [
          {k: torch.randn(v) for k, v in shapes.items()}
          for _ in range(num_samples)
      ]
  
file_path: pyro/tests/ops/test_streaming.py
imports: import functools
  import pytest
  import torch
  from pyro.ops.streaming import CountMeanStats, CountMeanVarianceStats, CountStats, StackStats, StatsOfDict
  from tests.common import assert_close
label_desc: Use image node_id 1 for calling a global function with example usage: generate_data(num_samples) and returns: unknown