code: def sampler(model, sampler_config):
      return VAMPSampler(model=model, sampler_config=sampler_config)
  
file_path: benchmark_VAE/tests/test_vamp_sampler.py
imports: import os
  import numpy
  import pytest
  import torch
  from pythae.models import VAMP, VAMPConfig
  from pythae.pipelines import GenerationPipeline
  from pythae.samplers import NormalSampler, NormalSamplerConfig, VAMPSampler, VAMPSamplerConfig
label_desc: Use image node_id 4 for calling a global function with example usage: sampler(model, sampler_config) and returns: VAMPSampler