code: def generate_data():
      return Data(
          z=torch.randint(1, 10, (20,)),
          pos=torch.randn(20, 3),
      )
  
file_path: pytorch_geometric/test/nn/models/test_schnet.py
imports: import pytest
  import torch
  from torch_geometric.data import Batch, Data
  from torch_geometric.nn import SchNet
  from torch_geometric.nn.models.schnet import RadiusInteractionGraph
  from torch_geometric.testing import is_full_test, withPackage
label_desc: Use image node_id 1 for calling a global function with example usage: generate_data() and returns: Data